scala – Spark SQL – IN子句

发布时间：2020-12-16 09:00:58 所属栏目：安全来源：网络整理

导读：我想在DataFrame中为具有多个值的列添加where条件. 例如,它为单值工作. df.where($"type".==="type1" $"status"==="completed"). 如何为同一列添加多个值,如下所示. df.where($"type" IN ("type1","type2") $"status" IN ("completed","inprogress") 解决方

我想在DataFrame中为具有多个值的列添加where条件.

例如,它为单值工作.

df.where($"type".==="type1" && $"status"==="completed").

如何为同一列添加多个值,如下所示.

df.where($"type" IN ("type1","type2") && $"status" IN ("completed","inprogress")

解决方法

你正在寻找的方法是：

import sqlContext.implicits._
df.where($"type".isin("type1","type2") and $"status".isin("completed","inprogress"))

通常,你想做这样的事情

val types = Seq("type1","type2")
val statuses = Seq("completed","inprogress")

df.where($"type".isin(types:_*) and $"status".isin(statuses:_*))

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!