Python Pandasで既存のCSVファイルから複数のCSVファイルを作成する方法

以下のようなCSVファイルが存在すると仮定します。

SalesRecords.csv

この既存のCSVファイルから、3つのCSVファイルを生成する必要があります。生成されるファイルは「Car」列の車名ごとに分割され、BMW.csv、Lexus.csv、Jaguar.csv という名前で保存されます。

ステップ1: 入力CSVファイルを読み込む

まず、read_csv() メソッドを使って、入力元となるCSVファイル(SalesRecords.csv)を読み込みます。

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

ステップ2: groupby() で車名ごとにCSVファイルを生成する

「Car」列の値(車名)を基準にデータをグループ化し、それぞれを個別のCSVファイルとして出力するには、groupby() メソッドを使用します。

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

ここでは f文字列 を使って、車名をそのままファイル名に反映させています。また、index=False を指定することで、不要なインデックス列がCSVファイルに出力されるのを防げます。

サンプルコード

以下は、これまでの手順をすべてまとめた完全なコードです。

import pandas as pd

# 入力CSVファイルを読み込む
 dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesRecords.csv")
print("\nInput CSV file = \n", dataFrame)

# 「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      BMW         10/17/2020
2         5      BMW         10/22/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ファイルが正しく生成されていることが確認できます。これらのファイルはプロジェクトディレクトリ内に自動的に作成されます。今回の例では PyCharm IDE 上で実行しているため、生成された3つのCSVファイルのパスは以下のようになっています。

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