如何在单元格列表中包含 object 的 select 行

Question

这是我的问题的 MWE：

import pandas as pd
d = {"A": [[1,2,3],[2,3]], "B": ["Hey", "Test"]}
df = pd.DataFrame.from_dict(d)
df.loc[1 in df["A"]]

投掷

KeyError: 'True: boolean label 不能在没有 boolean 索引的情况下使用'

我试了一个 function 像这样：

t = lambda x, y: y in x 
df.loc[t(df["A"],1)]

但错误仍然存在。 如何使用in运算符，或者实际上是任何不比较两个字符串的运算符？ 我有一种感觉，问题是我在我的单元格中使用列表，而 pandas 不喜欢我这样做，但我不知道，我对这个库不是很有经验。

Answer 1

您需要使用apply来模仿预期的行为：

df[df['A'].apply(lambda x: 1 in x)]

output：

           A    B
0  [1, 2, 3]  Hey

Answer 2

另一种选择可能是groupby_any explode是否出现任何匹配项：

num = 1
msk = df['A'].explode().eq(num).groupby(level=0).any()
out = df[msk]

Output：

           A    B
0  [1, 2, 3]  Hey

Answer 3

您可以使用df.iloc[0] select 整个第一行：产生

A    [1, 2, 3]
B          Hey
Name: 0, dtype: object

您可以使用df['A'] select 整个第一列
产生：

0    [1, 2, 3]
1       [2, 3]
Name: A, dtype: object  

您可以 select 通过df['A'].iloc[0]仅在“A”列的第一行产生：

[1, 2, 3]  

然后，您可以通过df['A'].iloc[0][0]对第一行 Col A 条目进行切片以获取第一个值
产生：

1