Pyspark数据帧丢弃列问题

Question

我试图从数据帧中删除两列，但我面临一个错误

**Error:**
drop() takes 2 positional arguments but 3 were given

***Code:***
 excl_columns= row['exclude_columns'].split(',')
 df=df.drop(*excl_columns)

#print(excl_columns)
#['year_of_birth', 'ethnicity']

Answer 1

这是应该工作的一种方式：

excl_columns = row['exclude_columns'].split(',')
df.select([c for c in df.columns if c not in excl_columns])