使用 Java 检查 Spark Dataframe 中是否存在列

Question

根据How do I detect if a Spark DataFrame has a column ，有一个 function 像df.columns.contains("column-name-to-check")可以检查列是否存在。 我四处搜索，并没有在 Java Spark 中找到类似的 function。 有谁知道Java中是否有类似的？

Answer 1

选项之一：

Arrays.asList(df.columns()).contains("column-name-to-check")

Answer 2

在 Java API 中， columns返回String[] 。 因此，我猜你可以使用java.util.Arrays （因为Java 8 ）：

String yourColumn = "columnName";
String[] columns = df.columns();
boolean contains = Arrays.stream(columns).anyMatch(yourColumn::equals);