Pandas Practice Set-1: Read rows in which the ‘cut’ is ‘Premium’, column ‘color’ of diamonds DataFrame
54. Read Rows Where 'cut' is 'Premium', Column 'color'
Write a Pandas program to read rows in which the ‘cut’ is ‘Premium’, column ‘color’ of diamonds DataFrame.
Sample Solution:
Python Code:
import pandas as pd
diamonds = pd.read_csv('https://raw.githubusercontent.com/mwaskom/seaborn-data/master/diamonds.csv')
print("Original Dataframe:")
print(diamonds.head())
print("\nRows in which the ‘cut’ is ‘Premium’, column ‘color’:")
print(diamonds.loc[diamonds.cut=='Premium', 'color'])
Sample Output:
Original Dataframe:
carat cut color clarity depth table price x y z
0 0.23 Ideal E SI2 61.5 55.0 326 3.95 3.98 2.43
1 0.21 Premium E SI1 59.8 61.0 326 3.89 3.84 2.31
2 0.23 Good E VS1 56.9 65.0 327 4.05 4.07 2.31
3 0.29 Premium I VS2 62.4 58.0 334 4.20 4.23 2.63
4 0.31 Good J SI2 63.3 58.0 335 4.34 4.35 2.75
Rows in which the ‘cut’ is ‘Premium’, column ‘color’:
1 E
3 I
12 F
14 E
15 E
26 I
45 F
53 E
54 D
56 J
61 D
64 I
68 G
69 E
72 H
73 H
85 G
86 H
87 H
88 H
89 I
99 H
101 E
103 G
106 G
112 I
116 F
125 F
126 H
135 E
..
53827 E
53834 E
53835 G
53836 D
53853 F
53856 D
53857 D
53859 E
53860 D
53861 D
53862 E
53867 G
53868 F
53871 F
53872 H
53877 F
53880 G
53888 G
53901 E
53904 E
53905 E
53910 E
53911 E
53912 F
53918 I
53928 E
53930 E
53931 F
53934 D
53938 H
Name: color, Length: 13791, dtype: object
For more Practice: Solve these Related Problems:
- Write a Pandas program to filter the diamonds DataFrame for rows where the 'cut' equals 'Premium' and display only the 'color' column.
- Write a Pandas program to use loc to select rows with 'cut' equal to 'Premium' and then extract the 'color' column values.
- Write a Pandas program to filter diamonds for a 'Premium' cut and output the unique colors present.
- Write a Pandas program to retrieve rows where the diamond cut is Premium and then count the frequency of each 'color' in those rows.
Go to:
PREV : Read Rows 0 Through 2 (Inclusive), Columns 'color' Through 'price' (Inclusive).
NEXT : Read Rows in Positions 0 and 1, Columns in Positions 0 and 3.
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?
