python - .loc 用于 Pandas 中的分类索引

Question

我尝试使用 .loc 访问基于 CategoricalIndex 的 Pandas 数据帧的行，但收到​​ TypeError。最小的非工作示例是

import pandas as pd

df = pd.DataFrame({'foo': rand(3), 'future_index': [22, 13, 87]})
df['future_index'] = df['future_index'].astype('category')
df = df.set_index('future_index')

然后，在尝试访问与标签 13 对应的行时

df.loc[13]

我明白了

TypeError: cannot do label indexing on <class 'pandas.core.indexes.category.CategoricalIndex'> with these indexers [13] of <class 'int'>

尽管

13 in df.index

是真实。我知道我最终可以获得索引 13

df.index.get_loc(13)

但是，为什么上述更简单的方法不起作用？我错过了什么？

干杯。

Answer 1

对我来说:

print (df.loc[pd.CategoricalIndex([13])])
              foo
future_index     
13              2

但是如果如上所述转换为 str EdChum它工作得很好:

df = pd.DataFrame({'foo': [1,2,3], 'future_index': [22, 13, 87]})
df['future_index'] = df['future_index'].astype(str).astype('category')
df = df.set_index('future_index')
print (df)

print (df.loc['13'])
foo    2
Name: 13, dtype: int64