如何在Dataframe的list-type Column 2中查找/比較Column 1的integer？

Question

我有一個如下所示的 dataframe - 我想測試列Number是否在每行記錄的列號List of Numbers 。

最終我希望得到如下所示的Result列：

Python Pandas 有沒有更好的方法得到預期的結果？ 謝謝！

Answer 1

in語句中使用列表理解：

df['Result'] = [b in a for a, b in df[['List of Numbers','Number']].values]

與zip類似的想法：

df['Result'] = [b in a for a, b in zip(df['List of Numbers'],df['Number'])]

或解決方案DataFrame.apply ：

df['Result'] = df.apply(lambda x: x['Number'] in x['List of Numbers'], axis=1)

編輯：更改df['Result']以屏蔽上述任何解決方案boolean indexing過濾：

mask = df.apply(lambda x: x['Number'] in x['List of Numbers'], axis=1)
df1 = df[mask]