Computer >> Máy Tính >  >> Lập trình >> Python

Python Pandas- Tạo nhiều tệp CSV từ tệp CSV hiện có

Giả sử sau đây là tệp CSV của chúng tôi -

SalesRecords.csv

Python Pandas- Tạo nhiều tệp CSV từ tệp CSV hiện có

Và chúng ta cần tạo 3 tệp excel từ tệp CSV hiện có ở trên. 3 tệp CSV phải dựa trên tên Xe, tức là BMW.csv, Lexus.csv và Jaguar.csv.

Đầu tiên, hãy đọc tệp CSV đầu vào của chúng tôi, tức là SalesRecord.csv -

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

Sử dụng groupby () để tạo CSV trên cơ sở Tên xe trong cột Xe -

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

Ví dụ

Sau đây là mã -

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"))

Đầu ra

Điều này sẽ tạo ra kết quả sau -

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

Như bạn có thể thấy ở trên, 3 tệp CSV đã được tạo. Các tệp CSV này được tạo trong thư mục dự án. Trong trường hợp của chúng tôi, sau đây là đường dẫn của tất cả ba tệp CSV, vì chúng tôi đang chạy trên PyCharm IDE -

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