如何在Java中使用Column.isin？

Question

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

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

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

传递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()));