python - 将特定列的值与不同文件进行比较

Question

我有一个名为 data 的数据框，它有 2 列，如下所示:

color  fruitN
red    apple
yellow orange
blue   banana
green  avocado

另外，我有 3 个不同的文件，每一列；每个包含水果列表如下:

file1
akee
apricot
avocado

file2
avocado
bilberry
banana
blackberry

file3
blackberry
coconut
cranberry

我需要创建另一个名为 type 的数据框，其行数与 data= 4 相同，列数与文件数相同 = 3

我需要检查数据帧data的fruitN列中的每个水果是否存在于第一个文件中，将1放入第一行第一列中，然后继续检查文件2和文件3。因此，输出数据帧 type 应该是这样的:

c1 c2 c3
0  0  0
0  0  0
0  1  0
1  1  0

然后，与data连接起来，如下所示:

color  fruitN c1 c2 c3
red    apple  0  0  0
yellow orange 0  0  0
blue   banana 0  1  0
green  avocado 1 1  0

我是 python 初学者，因此我们将不胜感激。

Answer 1

使用isin

l=[f1,f2,f3]
for x,y in enumerate(l):
    df['c'+str(x+1)]=df.fruitN.isin(y.iloc[:,0].tolist()).astype(int)

df
Out[144]: 
    color   fruitN  c1  c2  c3
0     red    apple   0   0   0
1  yellow   orange   0   0   0
2    blue   banana   0   1   0
3   green  avocado   1   1   0