创建一个函数来遍历现有数据框并将值添加到新列

Question

我有一个包含以下信息的数据框：

import pandas as pd
df = pd.DataFrame(data={'School':['A', 'B', 'C', 'D'],
                        'Year':[2009, 2010, 2011, 2012})

df
School   Year
A        2009
B        2010
C        2011
D        2012

我还有一个单独的学校列表（例如df2 = pd.DataFrame(data={'School':['A', 'B']}) ）存储在不同的数据帧中。

我想创建一个函数，该函数将循环遍历学校列，如果学校在列表中，则将 True 或 false 添加到新列。

我该怎么做呢？

Answer 1

您可以使用熊猫系列isin方法：

df['in_list'] = df['School'].isin(df2['School'])