columns

Načte názvy všech sloupců v DataFrame seznamu. Pořadí názvů sloupců v seznamu odráží jejich pořadí v datovém rámci.

Návraty

list

Příklady

Načtěte názvy sloupců datového rámce.

df = spark.createDataFrame(
    [(14, "Tom", "CA"), (23, "Alice", "NY"), (16, "Bob", "TX")],
    ["age", "name", "state"]
)
df.columns
# ['age', 'name', 'state']

Názvy sloupců slouží k promítání konkrétních sloupců.

selected_cols = [col for col in df.columns if col != "age"]
df.select(selected_cols).show()
# +-----+-----+
# | name|state|
# +-----+-----+
# |  Tom|   CA|
# |Alice|   NY|
# |  Bob|   TX|
# +-----+-----+

Zkontrolujte, jestli v datovém rámci existuje konkrétní sloupec.

"state" in df.columns
# True
"salary" in df.columns
# False