python - 从分类中提取 value_counts() 之后的索引作为数组

Question

自从我开始categorical在 pandas 中使用类型以来，我有一段特定的代码不起作用：（为方便起见，我将其形成为测试）：

import pandas as pd
import numpy as np
df = pd.DataFrame({"A": ["a", "b", "c", "a"]})

df["B"] = df["A"].astype('category')
original_grouped=df["A"].value_counts()
assert(isinstance( original_grouped.index.values, np.ndarray))

b_grouped=df["B"].value_counts()
print(type( b_grouped.index.values))
assert(isinstance( b_grouped.index.values, np.ndarray))

我怎样才能转换b_grouped.index.values成一个np.ndarray？经过一番搜索，我找不到和属性或设置方法来提取它

标签： pythonpandasnumpy

Answer 1

对我Series.to_numpy在 pandas 0.25.1 工作的我来说：

print(type( b_grouped.index.to_numpy()))
<class 'numpy.ndarray'>

assert(isinstance( b_grouped.index.to_numpy(), np.ndarray))