pyspark.sql.functions.array_contains

pyspark.sql.functions.array_contains(col, value)[source]

Collection function: returns null if the array is null, true if the array contains the given value, and false otherwise.

New in version 1.5.0.

Parameters
colColumn or str

name of column containing array

value :

value or column to check for in array

Examples

>>> df = spark.createDataFrame([(["a", "b", "c"],), ([],)], ['data'])
>>> df.select(array_contains(df.data, "a")).collect()
[Row(array_contains(data, a)=True), Row(array_contains(data, a)=False)]
>>> df.select(array_contains(df.data, lit("a"))).collect()
[Row(array_contains(data, a)=True), Row(array_contains(data, a)=False)]