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