Python – Create a Subset of columns using filter()

PythonServer Side ProgrammingProgramming

To create a subset of columns, we can use filter(). Through this, we can filter column values with similar pattern using like operator. At first, let us create a DataFrame with 3 columns −

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

Now, let us create a subset with multiple columns −

dataFrame[['Opening_Stock','Closing_Stock']]

Create a subset with similarly patterned names −

dataFrame.filter(like='Open')

Example

Following is the complete code −

import pandas as pd

dataFrame = pd.DataFrame({"Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],"Opening_Stock": [300, 700, 1200, 1500],"Closing_Stock": [200, 500, 1000, 900]})

print"DataFrame...
",dataFrame print"
Displaying a subset using indexing operator:
",dataFrame[['Product']] print"
Displaying a subset with multiple columns:
",dataFrame[['Opening_Stock','Closing_Stock']] print"
Displaying a subset with similarly patterned names:
",dataFrame.filter(like='Open')

Output

This will produce the following output −

DataFrame...
   Closing_Stock   Opening_Stock    Product
0           200             300     SmartTV
1           500             700  ChromeCast
2          1000            1200     Speaker
3           900            1500    Earphone

Displaying a subset using indexing operator:
      Product
0     SmartTV
1  ChromeCast
2     Speaker
3    Earphone

Displaying a subset with multiple columns:
   Opening_Stock   Closing_Stock
0         300             200
1         700             500
2        1200            1000
3        1500             900

Displaying a subset with similarly patterned names:
   Opening_Stock
0          300
1          700
2       1200
3 1500
raja
Updated on 14-Sep-2021 11:55:22

Advertisements