如何传递要在 pyspark 数据框中选择的列列表？

Question

我有列表列名。

columns = ['home','house','office','work']

我想将该列表值作为“选择”数据框中的列名称传递。

我试过了...

df_tables_full = df_tables_full.select('time_event','kind','schema','table',columns)

但我在下面收到错误..

TypeError: Invalid argument, not a string or column: ['home', 'house', 'office',
'work'] of type <class 'list'>. For column literals, use 'lit', 'array', 'struct' 
or 'create_map' function.

你能有什么想法吗？谢谢你们！

Answer 1

使用*before columns取消嵌套列列表并使用 in .select。

columns = ['home','house','office','work']

#select the list of columns
df_tables_full.select('time_event','kind','schema','table',*columns).show()

df_tables_full = df_tables_full.select('time_event','kind','schema','table',*columns)