scala - 如果值包含在 scala spark 的列表中，如何过滤行？

Question

例如，我想使用列表过滤 spark 数据框

var lisst=List(1,2,34)
df.filter(col("id).isin(lisst))

报错

Unsupported literal type classscala.collection.immutable.$colon$colon 
List(1,2,34)

我尝试使用 Seq 和 Set 并得到同样的错误。

Answer 1

你可以使用isin函数如下

var lisst=List(1,2,34)
 df.filter(col("id").isin(lisst :_*))

希望这对您有所帮助!