Как я могу передать список столбцов для выбора в pyspark dataframe?

Question

У меня есть имена столбцов списка.

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

, и я хотел бы передать значения этого списка в качестве имени столбцов в кадре данных "select".

Я пробовал это ...

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

Используйте * перед columns для удаления списка столбцов и используйте в .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)