<p><strong>当<strong>pyspark中的</strong>时,可以使用<strong>&;</strong>(for and)和<strong>|</strong>(for or)创建多个条件。</p>
<p>注意:在<strong>pyspark中,t必须将每个组合成条件的表达式括在括号()中</p>
<pre><code>%pyspark
dataDF = spark.createDataFrame([(66, "a", "4"),
(67, "a", "0"),
(70, "b", "4"),
(71, "d", "4")],
("id", "code", "amt"))
dataDF.withColumn("new_column",
when((col("code") == "a") | (col("code") == "d"), "A")
.when((col("code") == "b") & (col("amt") == "4"), "B")
.otherwise("A1")).show()
</code></pre>
<p>在Spark Scala代码中(<strong>&;</strong>)或(<strong>|</strong>)条件可在<strong>when</strong>函数中使用</p>
<pre><code>//scala
val dataDF = Seq(
(66, "a", "4"), (67, "a", "0"), (70, "b", "4"), (71, "d", "4"
)).toDF("id", "code", "amt")
dataDF.withColumn("new_column",
when(col("code") === "a" || col("code") === "d", "A")
.when(col("code") === "b" && col("amt") === "4", "B")
.otherwise("A1")).show()
</code></pre>
<p>一、二、二、三</p>
<pre><code>Output:
+---+----+---+----------+
| id|code|amt|new_column|
+---+----+---+----------+
| 66| a| 4| A|
| 67| a| 0| A|
| 70| b| 4| B|
| 71| d| 4| A|
+---+----+---+----------+
</code></pre>
<p>此代码段是从<a href="https://sparkbyexamples.com/spark-when-otherwise-example/" rel="nofollow noreferrer">sparkbyexamples.com</a>复制的</p>