isin
Vérifiez si la valeur de la colonne est dans une liste de valeurs.
Syntaxe
Python
isin(*cols)
parameter
parameter | Type | Description |
|---|---|---|
| Valeurs | Liste de valeurs à vérifier |
Renvoie
Colonne (booléen)
Exemples
Exemple 1 : Filtrer les lignes avec des noms dans les valeurs spécifiées.
Python
df = spark.createDataFrame([(2, "Alice"), (5, "Bob"), (8, "Mike")], ["age", "name"])
df[df.name.isin("Bob", "Mike")].orderBy("age").show()
Output
# +---+----+
# |age|name|
# +---+----+
# | 5| Bob|
# | 8|Mike|
# +---+----+
Exemple 2 : filtrez les lignes avec des âges dans la liste spécifiée.
Python
df[df.age.isin([1, 2, 3])].show()
Output
# +---+-----+
# |age| name|
# +---+-----+
# | 2|Alice|
# +---+-----+
Exemple 3 : Filtrer les lignes dont les noms ne figurent pas dans les valeurs spécifiées.
Python
df[~df.name.isin("Alice", "Bob")].show()
Output
# +---+----+
# |age|name|
# +---+----+
# | 8|Mike|
# +---+----+
Exemple 4 : Utiliser un DataFrame comme sous-requête IN.
Python
df.where(df.age.isin(spark.range(6))).orderBy("age").show()
Output
# +---+-----+
# |age| name|
# +---+-----+
# | 2|Alice|
# | 5| Bob|
# +---+-----+