在 PySpark 数据框中添加一个可为空的列

在 Spark 中,文字列在添加时不可为空:

from pyspark.sql import SparkSession,functions as F
spark = SparkSession.builder.getOrCreate()

df = spark.createDataFrame([(1,)],['c1'])

df = df.withColumn('c2',F.lit('a'))

df.printSchema()
#  root
#   |-- c1: long (nullable = true)
#   |-- c2: string (nullable = false)

如何创建可为空的列?

gangzai981760123 回答:在 PySpark 数据框中添加一个可为空的列

我发现的最短方法 - 使用 when(似乎不需要 otherwise 子句):

df = df.withColumn('c2',F.when(F.lit(1).isNotNull(),F.lit('a')))

完整测试结果:

from pyspark.sql import SparkSession,functions as F
spark = SparkSession.builder.getOrCreate()

df = spark.createDataFrame([(1,)],['c1'])
df = df.withColumn('c2',F.lit('a')))

df.show()
#  +---+---+
#  | c1| c2|
#  +---+---+
#  |  1|  a|
#  +---+---+

df.printSchema()
#  root
#   |-- c1: long (nullable = true)
#   |-- c2: string (nullable = true)
本文链接:https://www.f2er.com/15238.html

大家都在问