Pandas - 如何比较两个系列并将两者中的值附加到列表中

Question

我已经编写了代码，其中有一个包含所有类型食物的数据框。然后，我使用 str.contains 将其拆分为水果和蔬菜系列。我编写了代码，将两个系列中常见的任何食物附加到列表中：

fruit = fruit_2.tolist()#converting the series to a list
veg = veg_2.tolist()#converting the series to a list

for x in range (len(fruit)):
    for y in range (len(veg)):
        if fruit[x] == veg[y]:
            both.append(fruit[x]) 
print(both)

这只是想知道是否有人有一个使用 pandas 并且不使用 for 循环的解决方案。 谢谢

Answer 1

试试这个：

fruit_2[fruit_2.isin(veg_2)]

这将为您提供通用元素。

Answer 2

你可以使用np.intersect1d

import numpy as np

# gives a numpy array (you can later convert to series or list)
both = np.intersect1d(fruit_2, veg_2)  

感谢answer