如何从Spark DataFrame中删除列表中给出的多个列名？

Question

我有一个动态列表,它是根据n的值创建的.

n = 3
drop_lst = ['a' + str(i) for i in range(n)]
df.drop(drop_lst)

但上述情况并不奏效.

注意:

我的用例需要一个动态列表.

如果我只是在没有列表的情况下执行以下操作

df.drop('a0','a1','a2')

如何使drop功能与列表一起使用？

Spark 2.2似乎没有这种功能.有没有办法让它不使用select()？

Answer 1

您可以使用*运算符将列表的内容作为参数传递给drop():

df.drop(*drop_lst)

Answer 2

您可以将列名称指定为逗号分隔列表，例如

df.drop("col1","col11","col21")