从 Pandas 数据框中的列表中返回匹配元素

Question

我有一个整数列表 (gtin) 和一个 pandas 数据框 (df)，其中一列包含每行中的一个列表 (df['gtins'])

我想在数据框 (df['gtin_match']) 中定义一个新列，它将数据框列表中的第一个匹配元素返回到独立列表。

我尝试将下面的代码与 apply 函数和 lambda 函数一起使用，但它返回 df['gtins'] 中的列表列表。

我该如何解决这个问题？

gtin = [1234, 456, 2005, 91]

df['gtin_match'] = df['gtins'].apply(lambda x: [x for i in range(0, len(x)) if int(x[i]) in gtin])

Answer 1

在您的理解中，您应该返回 x[i]，因为 x 本身就是列表对象，而您只想返回 int：

df['gtin_match'] = df['gtins'].apply(lambda x: [x[i] for i in range(0, len(x)) if int(x[i]) in gtin])