為關鍵字過濾 dataframe 列，從找到每個關鍵字的行中返回單獨的列值（名稱）

Question

如果有一個數據框，並且如果我在另一列中找到關鍵字，我想返回一列中的值。 所以下面如果我搜索apple ，我希望 output 是[a,b]

像這樣：

names words
a     apple
b     apple
c     pear

我想要一個列表： [a,b]

我找到了使用str.contains返回 boolean 值的方法，但不確定如何從同一行中的另一列中獲取值，這將為我命名。 一定有一個帖子我找不到如果有人可以指導我那里。

Answer 1

你可以做

list(df[df['words'].str.contains('apple')]['names'])

導致

['a', 'b']

1. df['words'].str.contains('apple')為條件構建一個 boolean pandas 系列
2. 上一行產生的系列用於過濾原始 dataframe df
3. 在上一行產生的 dataframe 中，選擇了“名稱”列
4. 在上一行產生的 dataframe 中，該列是一個列表

完整代碼：

import io
import pandas as pd
data = """
names words
a     apple
b     apple
c     pear
"""
df = pd.read_csv(io.StringIO(data), sep='\s+')

lst = list(df[df['words'].str.contains('apple')]['names'])


>>>print(lst)

['a', 'b']