pyspark.pandas.Series.where

Series.where(cond: pyspark.pandas.series.Series, other: Any = nan) → pyspark.pandas.series.Series

Replace values where the condition is False.

Parameters
condboolean Series

Where cond is True, keep the original value. Where False, replace with corresponding value from other.

otherscalar, Series

Entries where cond is False are replaced with corresponding value from other.

Returns
Series

Examples

>>> from pyspark.pandas.config import set_option, reset_option
>>> set_option("compute.ops_on_diff_frames", True)
>>> s1 = ps.Series([0, 1, 2, 3, 4])
>>> s2 = ps.Series([100, 200, 300, 400, 500])
>>> s1.where(s1 > 0).sort_index()
0    NaN
1    1.0
2    2.0
3    3.0
4    4.0
dtype: float64
>>> s1.where(s1 > 1, 10).sort_index()
0    10
1    10
2     2
3     3
4     4
dtype: int64
>>> s1.where(s1 > 1, s1 + 100).sort_index()
0    100
1    101
2      2
3      3
4      4
dtype: int64
>>> s1.where(s1 > 1, s2).sort_index()
0    100
1    200
2      2
3      3
4      4
dtype: int64
>>> reset_option("compute.ops_on_diff_frames")