将pyspark数据框的列转换为小写

Question

我在pyspark中有一个数据框，其中包含大写的列，如ID，COMPANY等等

我想将这些列名称设为id company，依此类推。根据要求，将所有列细菌转换为小写或大写。

我想这样做，列的数据类型保持不变。

我们怎么能这样做？

Answer 1

使用DataFrame中的columns字段

df = // load
for col in df.columns:
    df = df.withColumnRenamed(col, col.lower())

或者，正如@ zero323建议：

df.toDF(*[c.lower() for c in df.columns])