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

Question

我在pyspark中有一个数据帧，它有大写的列，比如ID、COMPANY等等

我想让这些列名变成idcompany等等。根据需要将所有列转换为小写或大写。

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

我们怎么能做到？

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])