Check if column exists in Spark Dataframe using Java

Question

According to How do I detect if a Spark DataFrame has a column, there is a function like df.columns.contains("column-name-to-check") which can check whether a column exists. I searched around and didn't find similar function in Java Spark. Does anybody know whether there's a similar one in Java?

Answer 1

One of the options:

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

Answer 2

In the Java API, columns returns String[]. Therefore, I guess you can use java.util.Arrays (since Java 8):

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