从数据帧索引筛选包含字符串模式的行

Question

我需要从Pandas数据框索引中过滤包含字符串模式的行。

我发现了下面的示例：如何从熊猫数据帧中筛选包含字符串模式的行，其中数据帧使用df[df[“col”].str.contains（）]进行过滤，这在本示例中效果很好。

df = pd.DataFrame({'vals': [1, 2, 3, 4], 'ids': [u'aball', u'bball', u'cnut', u'fball']})

在本例中，如果我将列id复制到索引，我可以使用df.index.str.contains（ball），这也可以正常工作。

但是，当我使用df时。指数str.contains（“示例”）在我的数据帧中不起作用。

我认为它不起作用，因为在我的数据帧中，返回的值不是数组（[True，False…，True]，dtype=bool），而是索引（[True，False…，True]，dtype='object'，length=667）。

如何重新编写代码以使其正常工作？

我不粘贴我的数据框，因为我从一个大的excel表中读取它。

非常感谢。

Answer 1

您应该确保索引是字符串。下面的示例产生一个错误。

# Test data
df = DataFrame([1,2,3,4], index=['foo', 'foo1', 'foo2', 1], columns=['value'])
df[df.index.str.contains('foo')]

将索引转换为str可以获得预期结果。

df.index = df.index.astype('str')
df[df.index.str.contains('foo')]

      value
foo       1
foo1      2
foo2      3