colonnes

Récupère les noms de toutes les colonnes dans le DataFrame sous forme de liste. L'ordre des noms de colonnes dans la liste reflète leur ordre dans le DataFrame.

Renvoie

list

Exemples

Récupérer les noms de colonnes d'un DataFrame.

Python

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

Utilisez les noms de colonnes pour projeter des colonnes spécifiques.

Python

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

Vérifier si une colonne spécifique existe dans un DataFrame.

Python

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