withField

新增或替換結構體欄位中的欄位。

語法

withField(fieldName, col)

參數

參數 類型 說明
fieldName str 要新增或替換欄位名稱
col 資料行 欄位值的欄位表達式

退貨

資料行

Examples

from pyspark.sql import Row
from pyspark.sql.functions import lit
df = spark.createDataFrame([Row(a=Row(b=1, c=2))])
df.withColumn('a', df['a'].withField('b', lit(3))).select('a.b').show()
# +---+
# |  b|
# +---+
# |  3|
# +---+
df.withColumn('a', df['a'].withField('d', lit(4))).select('a.d').show()
# +---+
# |  d|
# +---+
# |  4|
# +---+