spark sql 更改返回的datafram指定字段数据为随机字典值？

Question

val A = Array("已绑定", "未绑定")

  val random = new Random
  val result = A(random.nextInt(A.length))

val frame1 = frame.withColumn("dim4", when(col("dim4").isNull, A(random.nextInt(A.length))).otherwise(0))

dim4字段的值替换为数组里面的随机值，上面的写法不能进行随机，都是固定的一个值
求 怎么变成随机的值呢？

Answer 1

用 org.apache.spark.sql.functions.rand 做随机，比如：

import org.apache.spark.sql.functions.{rand, when}

df.withColumn("xx", when(rand() > 0.5, "1").otherwise("0"))