python - 使用不区分大小写的方式从行名中选择数据帧行(如 `grep -i` )

Question

我有一个如下所示的数据框:

In [1]: mydict = {"1421293_at Hdgfl1":[2.140412,1.143337,3.260313],
                  "1429877_at Lrriq3":[9.019368,0.874524,2.051820]}

In [3]: import pandas as pd

In [4]:  df = pd.DataFrame.from_dict(mydict, orient='index')

In [5]: df
Out[5]:
                          0         1         2
1421293_at Hdgfl1  2.140412  1.143337  3.260313
1429877_at Lrriq3  9.019368  0.874524  2.051820

我想要做的是使用不区分大小写的查询从行名称中选择行。例如给定查询“hdgfl1”它应该返回:

                                         0                1               2
1421293_at Hdgfl1                 2.140412          1.143337          3.260313

“hdgfl1”是“1421293_at Hdgfl1”的不区分大小写的查询。基本上等同于 grep -i。

有什么方法可以做到？

Answer 1

你可以这样做:

query = 'hdgfl1'
mask = df.index.to_series().str.contains(query, case=False)
df[mask]

另一种可能性是:

mask = df.reset_index()['index'].str.contains(query, case=False)

但这慢了 2 倍。