如何在Spark Java中使用isin中的DataFrame過濾器？

Question

我正在嘗試使用Java中的列表來過濾Spark DataFrame。

java.util.List<Long> selected = ....; 
DataFrame result = df.filter(df.col("something").isin(????)); 

問題是isin（...）方法接受Scala Seq或Scala varang。

傳入JavaConversions.asScalaBuffer（選中）也不起作用。

任何想法？

Answer 1

你可以使用這樣的東西。

df.filter(col("something").isin("valu1","value2") 

OR

val list = List("value1","value2") 
df.filter(col("something").isin(list: _*) 

在Java：

df.filter(col("something").isin(list.stream().toArray(String[]::new))))