Pyspark/Python 数据框多列不等于

Question

我在 pyspark 中有一个数据框。

df.columns

['id', 'name', 'date', 'city']

df1 = df[[i for i in list(df.columns) if i != 'date']]

当我执行 df1.columns 时，我得到了

['id', 'name', 'city']

test_df = df[[i for i in list(df.columns) if i != 'date' 'city']]

当我做 test_df.columns 时，我得到了

['id, 'name', 'date', 'city'] 

而不是

['id', 'name']

为什么会这样？对多列使用 != 的正确方法是什么。

我不想使用 df.select 方法。

Answer 1

test_df = df[[i for i in list(df.columns) if i not in ['date', 'city']]]