如何在Pandas中基于字符串列表为DataFrame新增条件列？

你遇到的ValueError是因为df['country'].isin(my_list)返回的是布尔值Series（每行对应一个True/False），而普通的if/else只能处理单个布尔值，无法直接作用于Series，所以报错。下面是几种可行的修复方案：

• 方案一：用numpy.where逐元素判断（推荐）
  直接利用numpy.where对Series进行批量判断，生成对应结果：

  import numpy as np
  my_list = ['AA', 'TR', 'NZ']
  df = df.assign(flag=np.where(df['country'].isin(my_list), 1, 0))
  

  如果需要字符串类型的'1'/'0'，把结果转成str即可：

  df = df.assign(flag=np.where(df['country'].isin(my_list), '1', '0'))
  

• 方案二：布尔值转整数（简洁高效）
  Pandas的布尔Series可以直接通过astype(int)转换为0/1的整数Series，True对应1，False对应0：

  df = df.assign(flag=df['country'].isin(my_list).astype(int))
  

  同样，要字符串类型的话再加一层astype(str)：

  df = df.assign(flag=df['country'].isin(my_list).astype(int).astype(str))
  

• 方案三：逐行apply处理（仅小数据集推荐）
  如果一定要用if/else逻辑，需要用apply把判断逻辑作用到每个单独的元素上（而非整个Series）：

  df = df.assign(flag=df['country'].apply(lambda x: '1' if x in my_list else '0'))
  

  注意：这种方法在大数据集上效率远低于前两种，仅适合小数据场景。

内容的提问来源于stack exchange，提问作者Merve