apache-spark - PySpark 如何从 Dataframe 架构的 StructType 对象解析和获取字段名称

Question

我已经从 Hive 表创建了 Dataframe 并想要检索字段/列名称。

>>>a=df.schema
>>>a
StructType(List(StructField(empid, IntegerType, true), StructField(empname,StringType, true)))

如何从此对象中检索字段名称(empid、empname)。

Answer 1

使用 pyspark.sql.types.StructType.fieldnames :

fieldNames()

Returns all field names in a list.

>>> struct = StructType([StructField("f1", StringType(), True)])
>>> struct.fieldNames()
['f1']