python - 将 DataFrame 中的新派生列从 bool 值转换为整数

Question

假设我有一个具有此架构的 DataFrame x:

xSchema = StructType([ \
    StructField("a", DoubleType(), True), \
    StructField("b", DoubleType(), True), \
    StructField("c", DoubleType(), True)])

然后我有了 DataFrame:

DataFrame[a :double, b:double, c:double]

我想要一个整数派生列。我能够创建一个 bool 列:

x = x.withColumn('y', (x.a-x.b)/x.c > 1)

我的新模式是:

DataFrame[a :double, b:double, c:double, y: boolean]

但是，我希望 y 列包含 0 表示 False，1 表示 True。

cast 函数只能对列进行操作，不能对DataFrame 进行操作，withColumn 函数只能对DataFrame 进行操作。如何添加新列并将其同时转换为整数？

Answer 1

您使用的表达式计算为列，因此您可以像这样直接转换:

x.withColumn('y', ((x.a-x.b) / x.c > 1).cast('integer')) # Or IntegerType()