重置Pandas

Question

我有一个列为分类的数据框。

我删除了所有具有类别的行。

如何确保结果数据框只包含那些存在的类别，并且不会将删除的类别保留在索引中？

Answer 1

df = pd.DataFrame({'color':np.random.choice(['Blue','Green','Brown','Red'], 50)})

df.color = df.color.astype('category')

df.color.head()

输出：

0     Blue
1    Green
2     Blue
3    Green
4    Brown
Name: color, dtype: category
Categories (4, object): [Blue, Brown, Green, Red]

从数据框和类别中删除Brown。

df = df.query('color != "Brown"')

df.color = df.color.cat.remove_categories('Brown')

df.color.head()

输出：

0     Blue
1    Green
2     Blue
3    Green
7      Red
Name: color, dtype: category
Categories (3, object): [Blue, Green, Red]