- Trending Categories
- Data Structure
- Networking
- RDBMS
- Operating System
- Java
- MS Excel
- iOS
- HTML
- CSS
- Android
- Python
- C Programming
- C++
- C#
- MongoDB
- MySQL
- Javascript
- PHP
- Physics
- Chemistry
- Biology
- Mathematics
- English
- Economics
- Psychology
- Social Studies
- Fashion Studies
- Legal Studies
- Selected Reading
- UPSC IAS Exams Notes
- Developer's Best Practices
- Questions and Answers
- Effective Resume Writing
- HR Interview Questions
- Computer Glossary
- Who is Who
Python - How to select a subset of a Pandas DataFrame
Let’s say the following are the contents of our CSV file opened in Microsoft Excel −
At first, load data from a CSV file into a Pandas DataFrame −
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")
To select a subset, use the square brackets. Mention the column in the brackets and fetch single column from the entire dataset −
dataFrame['Car']
Example
Following is the code −
import pandas as pd # Load data from a CSV file into a Pandas DataFrame dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv") print("\nReading the CSV file...\n",dataFrame) # displaying only a single column res1 = dataFrame['Car']; # displaying only a subset print("\nDisplaying only one column Car : \n",res1) # displaying two columns res2 = dataFrame[['Car','Units']]; # displaying another subset print("\nDisplaying two columns : \n",res2)
Output
This will produce the following output −
Reading the CSV file... Car Reg_Price Units 0 BMW 2500 100 1 Lexus 3500 80 2 Audi 2500 120 3 Jaguar 2000 70 4 Mustang 2500 110 Displaying only one column Car : 0 BMW 1 Lexus 2 Audi 3 Jaguar 4 Mustang Name: Car, dtype: object Displaying two columns : Car Units 0 BMW 100 1 Lexus 80 2 Audi 120 3 Jaguar 70 4 Mustang 110
Advertisements