Python pyspark DataFrame.withColumn用法及代碼示例

通過添加列或替換具有相同名稱的現有列來返回新的 DataFrame 。

列表達式必須是對此 DataFrame 的表達式；嘗試從其他 DataFrame 添加列將引發錯誤。

參數：

colName：str

字符串，新列的名稱。

col：Column

新列的 Column 表達式。

注意：

此方法在內部引入了投影。因此，多次調用它，例如，通過循環以添加多個列可能會生成可能導致性能問題甚至 StackOverflowException 的大計劃。為避免這種情況，請同時對多列使用 select() 。

例子：

>>> df.withColumn('age2', df.age + 2).collect()
[Row(age=2, name='Alice', age2=4), Row(age=5, name='Bob', age2=7)]