How to use where() in Pandas

How to use where() in Pandas

To use the where() method in Pandas, you can use it on a Pandas series or DataFrame to select elements from the series or DataFrame based on a specified condition. This method will return a new series or DataFrame containing only the elements that satisfy the specified condition.

Example

Here's an example of using the where() method in Pandas on a series:

import pandas as pd

# create a sample series
s = pd.Series([1, 2, 3, 4, 5])

# use the where() method to select elements
# from the series that are greater than 3
s_filtered = s.where(s > 3)

# display the result
print(s_filtered)

This will use the where() method to select elements from the series that are greater than 3, and return a new series containing only those elements. The resulting filtered series will be displayed to the console. The output will be:

0    NaN
1    NaN
2    NaN
3    4.0
4    5.0
dtype: float64

Conclusion

The where() method allows you to easily select elements from a Pandas series or DataFrame based on a specified condition. It returns a new series or DataFrame containing only the elements that satisfy the specified condition, which you can use to further manipulate the data in the series or DataFrame.