Проверьте, существует ли столбец в Spark Dataframe с использованием Java

Question

Согласно Как определить, есть ли в Spark DataFrame столбец , существует функция типа df.columns.contains("column-name-to-check"), которая может проверить, существует ли столбец. Я искал вокруг и не нашел подобную функцию в Java Spark. Кто-нибудь знает, есть ли подобный в Java?

Answer 1

В Java API columns возвращает String[]. Поэтому, я думаю, вы можете использовать java.util.Arrays (так как Java 8 ):

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

Answer 2

Один из вариантов:

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