pyspark:选择特定列及其位置

df.columns[n] 然后 select

df = spark.createDataFrame([[1,2], [3,4]], ['a', 'b'])

n

n = 1
df.select(df.columns[n]).show()
+---+                                                                           
|  b|
+---+
|  2|
|  4|
+---+

选择除列以外的所有列 以下内容:

n = 1

drop

df.drop(df.columns[n]).show()
+---+
|  a|
+---+
|  1|
|  3|
+---+

df.select(df.columns[:n] + df.columns[n+1:]).show()
+---+
|  a|
+---+
|  1|
|  3|
+---+