在完整数据框中查找excat字符串

Question

我正在dataframe中搜索一个字符串，并希望获得列标题列表，该excat字符串出现的位置。你知道吗

我的数据帧是：

df
ColumnA ColumnB ColumnC
apple  app       appl
banana  app       banana

现在我想搜索：apple，我应该返回：

ColumnA

如果我搜索banana，我应该得到：

ColumnA
ColumnC

如何获取列标题，字符串在数据帧中的位置？你知道吗

Answer 1

假设您正在查找与字符串匹配的匹配项，如标题所示，可以使用^{}检查与给定字符串的相等性，^{}检查哪些列至少包含一个匹配项，并对数据帧的列执行布尔索引：

df.columns[df.eq('apple').any(0)]
# Index(['ColumnA'], dtype='object')

---

df.columns[df.eq('banana').any(0)]
# Index(['ColumnA', 'ColumnC'], dtype='object')