在 Python 中为 Pandas 数据框提取行

Question

我已将一个简单的查询日志导入 Python 中的 Pandas 数据帧（参见图片），并且想知道最有效的方法是提取包含“关键字”列中包含的任何给定关键字的所有行的最有效方法。

我可以遍历数据帧 - 但感觉使用数组/掩码可能有更快的方法。

非常感谢任何帮助。

Answer 1

您可以使用str.contains，例如：

In [1]: df = pd.DataFrame([['abc', 1], ['cde', 2], ['efg', 3]])

In [2]: df
Out[2]:
     0  1
0  abc  1
1  cde  2
2  efg  3

In [3]: df[0].str.contains('c')
Out[3]:
0     True
1     True
2    False
Name: 0, dtype: bool

In [4]: df[df[0].str.contains('c')]
Out[4]:
     0  1
0  abc  1
1  cde  2