創建一個函數來遍歷現有數據框並將值添加到新列

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'])