通过遍历列表为 pandas 列赋值

Question

我有一个由三部分组成的问题 - 我有三个列表，我正在尝试编写一个循环来迭代其中两个列表，以根据匹配（或通过列表列表）为新列分配值）。当我将它分配给下面的列时，它只打印 else 值或最后一个条件中的任何数字。提前谢谢你。

1. 为什么会这样？

2. 如何格式化这个循环？

3. 有人可以编写一个更好的 for 循环来迭代 all_names，将 1 分配给 all_names[0] 的匹配项，将 2 分配给 all_names[1] 的匹配项吗？ - enumerate 是否有用？

   first_names = ['David', 'Dani', 'Declan']
   second_names = ['Katz', 'Foley', 'Stevenson']
   all_names = [first_names, second_names]
   
   for row in df['name']:
       for elem in firstname_list:
          if elem in row:
             df['name_num']= 1
       for elem in secondname_list:
          if elem in row:
             df['name_num']= 2
       else:
            df['name_num']= 3 
   

Answer 1

检查np.select：

df['name_num'] = np.select( (df['name'].isin(first_names), 
                             df['name'].isin(second_names) ),
                            (1,2), 3)