w3resource

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:


Previous:Write a Pandas program to read rows 0 through 2 (inclusive), columns ‘color’ through ‘price’ (inclusive) of diamonds DataFrame.
Next: Write a Pandas program to read rows in positions 0 and 1, columns in positions 0 and 3 of diamonds DataFrame.

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?



Follow us on Facebook and Twitter for latest update.