Python Pandas- สร้างไฟล์ CSV หลายไฟล์จากไฟล์ CSV ที่มีอยู่

สมมติว่าต่อไปนี้คือไฟล์ CSV ของเรา –

SalesRecords.csv

และเราต้องสร้างไฟล์ excel 3 ไฟล์จากไฟล์ CSV ที่มีอยู่ด้านบน ไฟล์ CSV 3 ไฟล์ควรอยู่บนพื้นฐานของชื่อรถยนต์ เช่น BMW.csv, Lexus.csv และ Jaguar.csv

ขั้นแรก อ่านไฟล์ CSV อินพุตของเรา เช่น SalesRecord.csv -

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

ใช้ groupby() เพื่อสร้าง 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

ดังที่คุณเห็นด้านบน มีการสร้างไฟล์ CSV 3 ไฟล์ ไฟล์ 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