给出数据框df
,如何通过索引检索列的名称?
例如,可以通过其名称获取列的索引,如下所示:df.columns.indexOf("col1")
,但我需要相反的事情。
答案 0 :(得分:5)
df.columns
按顺序为您提供每个列名的Array[String]
。
df.columns(index)
将为您提供index
答案 1 :(得分:1)
dataFrame.colums
给出列的列表。您可以按dataFrame.columns(index)
如果你有数据集,你可以通过
ds.schema.fields(index)
希望这有帮助