Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python Pandas - 데이터 프레임에서 행의 하위 집합 선택

<시간/>

행의 하위 집합을 선택하려면 조건을 사용하고 데이터를 가져옵니다.

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

Python Pandas - 데이터 프레임에서 행의 하위 집합 선택

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

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

"단위"가 100개 이상인 자동차 레코드, 즉 행의 하위 집합을 원한다고 가정해 보겠습니다. 이를 위해 -

를 사용하십시오.
dataFrame[dataFrame["단위"]> 100]

이제 "Reg_Price"가 100 미만인 자동차 레코드, 즉 행의 하위 집합을 원한다고 가정해 보겠습니다. 이를 위해 -

를 사용하십시오.
dataFrame[dataFrame["Reg_Price"] <3000]

예시

다음은 코드입니다 -

pandas를 pd로 가져오기# CSV 파일에서 Pandas DataFramedataFrame으로 데이터 로드 =pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")print("\nCSV 파일 읽기 ...\n",dataFrame)# 두 개의 열 표시res2 =dataFrame[['Reg_Price','Units']];print("\n두 개의 열 표시 :\n",res2)# 행의 하위 집합 선택print("\ nSelect cars with Units than 100:\n",dataFrame[dataFrame["Units"]> 100])# 행의 하위 집합 선택print("\nReg_Price가 3000 미만인 자동차 선택:\n",dataFrame[dataFrame[" Reg_Price"] <3000])

출력

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

CSV 파일 읽기 중... Car Reg_Price Units0 BMW 2500 1001 Lexus 3500 802 Audi 2500 1203 Jaguar 2000 704 Mustang 2500 110단 하나의 열만 표시 Car :Reg_Price Units0 00101 Lexus 3500 803 Jaguar 2000 704 Mustang 2500 110Car :Reg_Price Units0 01001 Lexus 3500 1203 Jaguar 2500 1101열만 표시 Car :Reg_Price Units0 0001 개체Select cars with Units than 100:Car Reg_Price Units2 Audi 2500 1204 Mustang 2500 110Reg_Price가 3000 미만인 자동차 선택:Car Reg_Price Units0 BMW 2500 1002 Audi 2500 1200 1002 Audi 2500 1203 Jaguar 5pre 20