pandas - 检查数据框中的列列表是否为 pandas

Question

我有一个数据框，如下所示。

Unit_ID     Type      Sector       Plot_Number       Rental
1           Home      se1          22                50
2           Shop      se1          26                80

从上面我需要编写函数来检查如下所示的列列表是否在数据框中。

如果列表为['Unit_ID', 'Sector', 'Usage_Type', 'Price']

预期输出:“Usage_Type”列和“Price”列不在数据框中。

如果列表为['Unit_ID', 'Sector' , 'Type', 'Plot_Number' ]

预期输出:列表中的所有内容都在数据框中

Answer 1

您可以尝试使用以下方法:

#For checking if the list of columns are actually 
#a subset of the dataframe columns or not , you can use:

def myf1(x,to_check):
    if not set(to_check).issubset(set(x.columns)):
       return f"{' and '.join(set(to_check).difference(x.columns))} are not available in the dataframe"
    return "All columns are available in the dataframe"

---

to_check = ['Unit_ID', 'Sector'] 
myf1(df,to_check)
#'All columns are available in the dataframe'

to_check = ['Unit_ID', 'Sector','XYZ'] 
myf1(df,to_check)    
#'XYZ are not available in the dataframe'