追加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.

那么，我该如何将这个类别添加到结尾呢？

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()来删除未使用的类别。

Answer 2

出现此错误是因为append是list的属性，而不是分类对象。

因此，要修复此错误，您需要通过执行以下操作将pandas.core.categorical.Categories转换为列表类型

sub_c = sub_c.tolist()

然后

sub_c.append(c.tolist()[-1])

我会将最后一个元素放入sub_c，但现在sub_c是一个列表。