检索列表中所有列 DataFrame 的名称。 列表中的列名称的顺序反映其在 DataFrame 中的顺序。
退货
list
示例
检索 DataFrame 的列名称。
df = spark.createDataFrame(
[(14, "Tom", "CA"), (23, "Alice", "NY"), (16, "Bob", "TX")],
["age", "name", "state"]
)
df.columns
# ['age', 'name', 'state']
使用列名投影特定列。
selected_cols = [col for col in df.columns if col != "age"]
df.select(selected_cols).show()
# +-----+-----+
# | name|state|
# +-----+-----+
# | Tom| CA|
# |Alice| NY|
# | Bob| TX|
# +-----+-----+
检查数据帧中是否存在特定列。
"state" in df.columns
# True
"salary" in df.columns
# False