python - 附加 pandas.core.categorical.Categories

Question

我正在尝试对每个第 n 个类别进行子集化，然后附加最后一个类别。

这是一个例子：

1）制作类别

import pandas as pd
import numpy as np
a = np.arange(0, 10, 1)
b = np.arange(0, 10, 1)
c = pd.cut(x = b.tolist(), bins = a, right = True, include_lowest = True)

2）子集类别

sub_c = c[0::4]

这就是麻烦，我想将最后一个类别附加到sub_c，例如：

sub_c.append(c[-1])但我得到了错误：'Categorical' object has no attribute append.

那么我如何将这个类别添加到最后呢？

标签： pythonpandas

Answer 1

您可以选择最后一个类别以及其他类别：

sub_c = c[list(range(0, c.size, 4)) + [-1]].remove_unused_categories()
#[(-0.001, 1.0], (3.0, 4.0], (7.0, 8.0], (8.0, 9.0]], ...

另一种选择是准备分类代码列表：

codes = np.concatenate([c[0::4].codes, c[[-1]].codes])
sub_c = c.from_codes(codes, c.categories).remove_unused_categories()
#[(-0.001, 1.0], (3.0, 4.0], (7.0, 8.0], (8.0, 9.0]], ....

不要忘记调用.remove_unused_categories()以删除未使用的类别。