如何使用列.Java中的isin

Question

我正在尝试使用Java中的列表过滤Spark DataFrame。

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

问题是isin(...)方法接受Scala Seq或变量。

通过JavaConversions.asScalaBuffer(selected)也不工作。

任何想法?

Answer 1

使用stream方法如下:

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

Answer 2

更短一点的版本是:

df.filter(col("something").isin(selected.toArray()));