Python Pandas- 从现有的 CSV 文件创建多个 CSV 文件

假设以下是我们的 CSV 文件 –

SalesRecords.csv

我们需要从上面的现有 CSV 文件生成 3 个 excel 文件。这 3 个 CSV 文件应基于车名，即 BMW.csv、Lexus.csv 和 Jaguar.csv。

首先，读取我们的输入 CSV 文件，即 SalesRecords.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 读取我们的输入 CS 文件
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesRecords.csv")
print("\nInput CSV file = \n", dataFrame)

# groupby 基于汽车名称在 Car 列上生成 CSV 文件
for (car), group in dataFrame.groupby(['Car']):
   group.to_csv(f'{car}.csv', index=False)

# 显示生成的 CSV 文件的值
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