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

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']})），存储在不同的数据框中。

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

我该怎么做？

Answer 1

您可以使用pandas系列isin方法：

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