How to Use where() in Numpy and Pandas (Python)

Channel:
Subscribers:
53,100
Published on ● Video Link: https://www.youtube.com/watch?v=erJ1ce2aHHY



Category:
Guide
Duration: 5:19
12,938 views
312


This video shows how to use the where() function in numpy and pandas to extract indices based on logical conditions and populate new columns of data based on elementwise logic. The np.where() function can perform a similar operation to the ifelse() function in R.

If you find this video useful, like, share and subscribe to support the channel!
► Subscribe: https://www.youtube.com/c/DataDaft?sub_confirmation=1


Code used in this Python Code Clip:

import numpy as np
import pandas as pd

import statsmodels.api as sm #(To access mtcars dataset)
mtcars = sm.datasets.get_rdataset("mtcars", "datasets", cache=True).data

mtcars.head()

# Extract indices that meet a condition
inds = np.where(mtcars.mpg > 22)

inds

# Perform operations across an array or column based on a condition

np.where(mtcars.mpg > 22, # Condition
"High MPG", # Value to set if condition is True
"Low MPG") # Value to set if condition is False

# Perform elementwise operations on an array or column

np.where(mtcars.mpg > 22, # Condition
mtcars.mpg, # Value to set if condition is True
mtcars.cyl) # Value to set if condition is False



* Note: YouTube does not allow greater than or less than symbols in the text description, so the code above will not be exactly the same as the code shown in the video! I will use Unicode large < and > symbols in place of the standard sized ones. .


⭐ Kite is a free AI-powered coding assistant that integrates with popular editors and IDEs to give you smart code completions and docs while you’re typing. It is a cool application of machine learning that can also help you code faster! Check it out here: https://www.kite.com/get-kite/?utm_medium=referral&utm_source=youtube&utm_campaign=datadaft&utm_content=description-only







Tags:
numpy where
np.where
np.where()
pandas where
pandas ifelse
python version of R ifelse
ifelse in python
ifelse() in python
numpy where()
pandas elementwise logic
numpy elementwise logic
python basics
pandas basics
numpy basics
pandas data manipulation
populating arrays in numpy
population columns in pandas
pandas logical index
pandas filter rows
pandas subset rows
python data analysis
python
pandas
numpy
learn python
pandas for beginners