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

Python Pandas - 기존 CSV 파일에서 여러 CSV 파일 만들기

<시간/>

다음이 CSV 파일이라고 가정해 보겠습니다. −

SalesRecords.csv

Python Pandas - 기존 CSV 파일에서 여러 CSV 파일 만들기

그리고 위의 기존 CSV 파일에서 3개의 Excel 파일을 생성해야 합니다. 3개의 CSV 파일은 자동차 이름(예:BMW.csv, Lexus.csv 및 Jaguar.csv)을 기반으로 해야 합니다.

먼저 입력 CSV 파일(SalesRecord.csv −

)을 읽습니다.
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")

Groupby()를 사용하여 Car 열의 Car 이름을 기반으로 CSV 생성 -

for (car), group in dataFrame.groupby(['Car']):
   group.to_csv(f'{car}.csv', index=False)

예시

다음은 코드입니다 -

import pandas as pd

# DataFrame to read our input CS file
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")
print("\nInput CSV file = \n", dataFrame)

# groupby to generate CSVs on the basis of Car names in Car column
for (car), group in dataFrame.groupby(['Car']):
   group.to_csv(f'{car}.csv', index=False)

#Displaying values of the generated CSVs
print("\nCSV 1 = \n", pd.read_csv("BMW.csv"))
print("\nCSV 2 = \n", pd.read_csv("Lexus.csv"))
print("\nCSV 3 = \n", pd.read_csv("Jaguar.csv"))

출력

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

Input CSV file =
   Unnamed: 0    Car   Date_of_Purchase
0         0      BMW         10/10/2020
1         1    Lexus         10/12/2020
2         2      BMW         10/17/2020
3         3   Jaguar         10/16/2020
4         4   Jaguar         10/19/2020
5         5      BMW         10/22/2020

CSV 1 =
   Unnamed: 0    Car   Date_of_Purchase
0         0      BMW         10/10/2020
1         2    Lexus         10/12/2020
2         5      BMW         10/17/2020

CSV 2 =
   Unnamed: 0    Car   Date_of_Purchase
0         1    Lexus         10/12/2020

CSV 3 =
   Unnamed: 0    Car   Date_of_Purchase
0         3   Jaguar         10/16/2020
1         4   Jaguar         10/19/2020

위에서 볼 수 있듯이 3개의 CSV 파일이 생성되었습니다. 이 CSV 파일은 프로젝트 디렉토리에서 생성됩니다. 우리의 경우 다음은 세 가지 CSV 파일 모두의 경로입니다. PyCharm IDE에서 실행 중이므로 -

C:\Users\amit_\PycharmProjects\pythonProject\BMW.csv
C:\Users\amit_\PycharmProjects\pythonProject\Jaguar.csv
C:\Users\amit_\PycharmProjects\pythonProject\Lexus.csv