Isin

Проверьте, находится ли значение столбца в списке значений.

Синтаксис

isin(*cols)

Параметры

Параметр	Тип	Описание
cols	values	Список значений для проверки

Возвраты

Столбец (логический)

Примеры

Пример 1. Фильтрация строк с именами в указанных значениях.

df = spark.createDataFrame([(2, "Alice"), (5, "Bob"), (8, "Mike")], ["age", "name"])
df[df.name.isin("Bob", "Mike")].orderBy("age").show()

# +---+----+
# |age|name|
# +---+----+
# |  5| Bob|
# |  8|Mike|
# +---+----+

Пример 2. Фильтрация строк с возрастом в указанном списке.

df[df.age.isin([1, 2, 3])].show()

# +---+-----+
# |age| name|
# +---+-----+
# |  2|Alice|
# +---+-----+

Пример 3. Фильтрация строк с именами, не указанными значениями.

df[~df.name.isin("Alice", "Bob")].show()

# +---+----+
# |age|name|
# +---+----+
# |  8|Mike|
# +---+----+

Пример 4. Использование кадра данных в качестве вложенного запроса IN.

df.where(df.age.isin(spark.range(6))).orderBy("age").show()

# +---+-----+
# |age| name|
# +---+-----+
# |  2|Alice|
# |  5|  Bob|
# +---+-----+