Python pandas 使用篩選條件選擇 dataframe 的子集

Question

我有一個 Pandas DataFrame 如下

enter image description here我想查詢列以找出每個名稱包含“X”的所有列。

示例 output 類似於：(John, O, P) 這里 O 和 P 是針對 John 的具有字符“X”的列 ID。 我嘗試使用 loc 在列上查詢 Dataframe，但沒有得到 output。請指導以獲取所需的 output。

Answer 1

這是一個使用pandas.DataFrame.apply和dict返回字典的命題，其中鍵是人名，值是滿足條件的列名（等於"X" ）

dico= dict(zip(df["Name"], df.eq("X").apply(lambda x: x.index[x].tolist(), axis=1)))

＃Output：

print(dico)

{'John': ['O', 'P'], 'Dave': ['M', 'P']}