Pyspark：选择特定列及其位置

我想知道如何在数据框中选择具有编号但不具有名称的特定列？

在 Pandas 中是这样的：

df = df.iloc[:,2]

这是可能的 ？

您始终可以使用以下方式获取列的名称df.columns[n]进而select it:

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

选择位置处的列n:

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

选择除列之外的所有内容n:

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|
+---+