Python - Pandas DataFrame의 하위 집합을 선택하는 방법

다음이 Microsoft Excel에서 열린 CSV 파일의 내용이라고 가정해 보겠습니다. -

먼저 CSV 파일에서 Pandas DataFrame으로 데이터를 로드합니다. -

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")

하위 집합을 선택하려면 대괄호를 사용합니다. 대괄호 안의 열을 언급하고 전체 데이터 세트에서 단일 열을 가져옵니다 -

dataFrame['Car']

예

다음은 코드입니다 -

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)

출력

이것은 다음과 같은 출력을 생성합니다 -

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