在Python中基于两行之间的匹配创建一个新列[0,1]

Question

我正在尝试将多个列表或数据框架与一个大型基础数据框架进行比较。然后对于任何匹配，我想添加一个列存储1 = match或0 = No match

df = pd.DataFrame({'Name':['A','B','C','D'], 'ID' : ['5-6','6-7','8-9','7']})
list1 = ['5-6','8-9']
list2 = ['7','4-3']

由于我试图匹配的值包括'-'，它们被计数为字符串。我可以生成一个匹配值的列表，但是如果我添加它们，它们都是0

def f(rows):
for i in df['ID']:
for j in list1:
if i == j:
val = 1
else:
val = 0
return val
df['Answer']= df.apply(f,axis=1)

,

for i in df['ID']:
for j in  list1:
if i == j:
print (i)

查找所有匹配值。

提前感谢!

Answer 1

你已经循环了.apply，所以你可以省略循环，测试是使用in作为列表的成员:

def f(rows):
if rows['ID'] in list1:
val = 1
else:
val = 0
return val
df['Answer']= df.apply(f,axis=1)
print (df)
Name   ID  Answer
0    A  5-6       1
1    B  6-7       0
2    C  8-9       1
3    D    7       0

更简单的是使用指定列的lambda函数:

df['Answer']= df['ID'].apply(lambda x: 1 if x in list1 else 0)

或:

df['Answer']= df['ID'].apply(lambda x: int(x in list1))

Answer 2

可以使用DataFrame。这里用Isin代替loop

df['Answer'] = df['ID'].isin(list1).astype(int)
Name    ID  Answer
0   A       5-6 1
1   B       6-7 0
2   C       8-9 1
3   D       7   0