使用关键字 Pandas 数据框 Python 进行过滤

Question

我在数据框中存储了数据。现在我想使用关键字过滤掉特定列中的数据。例如：让我的数据框为：

    customer    product         state
0   shyam       pen drive       delhi
1   suresh      hard disk       punjab
2   shanu       USB pen drive   haryana
3   rohan       pen drive 16gb  uttarpradesh

现在我想用仅包含 'pen' 的值过滤 'product' 列中的数据框。所以结果应该是这样的：

    customer    product         state
0   shyam       pen drive       delhi
2   shanu       USB pen drive   haryana
3   rohan       pen drive 16gb  uttarpradesh

我们将不胜感激。

Answer 1

使用这个：

df[df['product'].str.contains('pen')]

输出：

  customer         product         state
0    shyam       pen drive         delhi
2    shanu   USB pen drive       haryana
3    rohan  pen drive 16gb  uttarpradesh

.str，字符串访问器，可能会很慢，使用列表理解以获得更好的性能：

df[['pen' in i for i in df['product']]]

Answer 2

df2 = df.loc[df.product.str.contains('pen')]

使用关键字 Pandas 数据框 Python 进行过滤

Filter using keyword Pandas Dataframe Python

filtering

python-3.x

pandas