根据条件在数据框熊猫中创建列

Question

我有两个清单

A = ['a','b','c','d','e']
B = ['c','e']

具有列的数据框

      A
  0   a
  1   b
  2   c
  3   d
  4   e

我希望为B中的元素与A匹配的行创建一个附加列。

      A  M
  0   a
  1   b
  2   c match
  3   d
  4   e match

Answer 1

您可以使用loc或numpy.where和条件与isin ：

df.loc[df.A.isin(B), 'M'] = 'match'
print (df)
   A      M
0  a    NaN
1  b    NaN
2  c  match
3  d    NaN
4  e  match

要么：

df['M'] = np.where(df.A.isin(B),'match','')
print (df)

   A      M
0  a       
1  b       
2  c  match
3  d       
4  e  match