Pandas: Keep only the rows with at least 2 non-NA values
8. Keep Rows with at Least 2 NaN Values
Write a Pandas program to keep the rows with at least 2 NaN values in a given DataFrame.
Test Data:
ord_no purch_amt ord_date customer_id
0 NaN NaN NaN NaN
1 NaN 270.65 2012-09-10 3001.0
2 70002.0 65.26 NaN 3001.0
3 NaN NaN NaN NaN
4 NaN 948.50 2012-09-10 3002.0
5 70005.0 2400.60 2012-07-27 3001.0
6 NaN 5760.00 2012-09-10 3001.0
7 70010.0 1983.43 2012-10-10 3004.0
8 70003.0 2480.40 2012-10-10 3003.0
9 70012.0 250.45 2012-06-27 3002.0
10 NaN 75.29 2012-08-17 3001.0
11 NaN NaN NaN NaN
Sample Solution:
Python Code :
import pandas as pd
import numpy as np
pd.set_option('display.max_rows', None)
#pd.set_option('display.max_columns', None)
df = pd.DataFrame({
'ord_no':[np.nan,np.nan,70002,np.nan,np.nan,70005,np.nan,70010,70003,70012,np.nan,np.nan],
'purch_amt':[np.nan,270.65,65.26,np.nan,948.5,2400.6,5760,1983.43,2480.4,250.45, 75.29,np.nan],
'ord_date': [np.nan,'2012-09-10',np.nan,np.nan,'2012-09-10','2012-07-27','2012-09-10','2012-10-10','2012-10-10','2012-06-27','2012-08-17',np.nan],
'customer_id':[np.nan,3001,3001,np.nan,3002,3001,3001,3004,3003,3002,3001,np.nan]})
print("Original Orders DataFrame:")
print(df)
print("\nKeep the rows with at least 2 NaN values of the said DataFrame:")
result = df.dropna(thresh=2)
print(result)
Sample Output:
Original Orders DataFrame:
ord_no purch_amt ord_date customer_id
0 NaN NaN NaN NaN
1 NaN 270.65 2012-09-10 3001.0
2 70002.0 65.26 NaN 3001.0
3 NaN NaN NaN NaN
4 NaN 948.50 2012-09-10 3002.0
5 70005.0 2400.60 2012-07-27 3001.0
6 NaN 5760.00 2012-09-10 3001.0
7 70010.0 1983.43 2012-10-10 3004.0
8 70003.0 2480.40 2012-10-10 3003.0
9 70012.0 250.45 2012-06-27 3002.0
10 NaN 75.29 2012-08-17 3001.0
11 NaN NaN NaN NaN
Keep the rows with at least 2 NaN values of the said DataFrame:
ord_no purch_amt ord_date customer_id
1 NaN 270.65 2012-09-10 3001.0
2 70002.0 65.26 NaN 3001.0
4 NaN 948.50 2012-09-10 3002.0
5 70005.0 2400.60 2012-07-27 3001.0
6 NaN 5760.00 2012-09-10 3001.0
7 70010.0 1983.43 2012-10-10 3004.0
8 70003.0 2480.40 2012-10-10 3003.0
9 70012.0 250.45 2012-06-27 3002.0
10 NaN 75.29 2012-08-17 3001.0
For more Practice: Solve these Related Problems:
- Write a Pandas program to select rows from a DataFrame that contain at least two missing values.
- Write a Pandas program to filter the DataFrame so that only rows with a minimum of two NaN entries are retained.
- Write a Pandas program to use dropna() with a threshold parameter to keep rows with at least two missing elements.
- Write a Pandas program to count NaNs per row and then return only those rows with two or more missing values.
Go to:
Previous: Write a Pandas program to drop the rows where all elements are missing in a given DataFrame.
Next: Write a Pandas program to drop those rows from a given DataFrame in which specific columns have missing values.
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
What is the difficulty level of this exercise?
Test your Programming skills with w3resource's quiz.