Python PySpark DataFrame toDF方法用法及代碼示例

例子

考慮以下PySpark DataFrame：

df = spark.createDataFrame([["Alex", 20], ["Bob", 30]], ["name", "age"])
df.show()



+----+---+
|name|age|
+----+---+
|Alex| 20|
| Bob| 30|
+----+---+

在 PySpark 中按特定順序排列列

要從 age 首先排列列，然後從 name 第二排列列：

df.toDF("age", "name").show()



+----+----+
| age|name|
+----+----+
|Alex|  20|
| Bob|  30|
+----+----+

請注意，如果新的 DataFrame 的列與原始 DataFrame 不匹配，則會拋出錯誤：

df.toDF("age").show()



IllegalArgumentException: requirement failed: The number of columns doesn't match.
Old column names (2): name, age
New column names (1): age

按字母順序排列 PySpark 中的列

要按字母順序排列列：

df.toDF(*sorted(df.columns)).show()



+----+----+
| age|name|
+----+----+
|Alex|  20|
| Bob|  30|
+----+----+

這裏：

• sorted(~) 按字母順序返回列標簽。

• * 用於將列表轉換為位置參數。