withColumn

Vrátí nový datový rámec přidáním sloupce nebo nahrazením existujícího sloupce se stejným názvem.

Syntaxe

withColumn(colName: str, col: Column)

Parametry

Parameter	Typ	Description
colName	str	řetězec, název nového sloupce.
col	Sloupec	a Column expression for the new column.

Návraty

DataFrame: Datový rámec s novým nebo nahrazeným sloupcem.

Poznámky

Tato metoda zavádí projekci interně. Proto volání několikrát, například prostřednictvím smyček, aby bylo možné přidat více sloupců může generovat velké plány, které můžou způsobit problémy s výkonem a dokonce i StackOverflowException. Chcete-li tomu zabránit, použijte select s více sloupci najednou.

Příklady

df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
df.withColumn('age2', df.age + 2).show()
# +---+-----+----+
# |age| name|age2|
# +---+-----+----+
# |  2|Alice|   4|
# |  5|  Bob|   7|
# +---+-----+----+