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

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

销售记录.csv

我们需要从上述现有的 CSV 文件中生成 3 个 excel 文件。3 个 CSV 文件应基于汽车名称,即 BMW.csvLexus.csv和 Jaguar.csv。

首先,阅读我们的输入 CSV 文件,即SalesRecord.csv-

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

用于groupby()根据 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

以上是 Python Pandas - 从现有的 CSV 文件创建多个 CSV 文件 的全部内容, 来源链接: utcz.com/z/363217.html

回到顶部