withField

在结构列中添加或替换字段。

Syntax

withField(fieldName, col)

参数

参数 类型 Description
fieldName str 要添加或替换的字段的名称
col 字段值的列表达式

退货

示例

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|
# +---+