pandasを使ったCSVファイルの分割方法

Question

やりたいことは、画像の通りです。

※要素1の左側は”カテゴリ”です

以下プログラムを書きました。

# -*- coding: utf-8 -*-
import numpy as numpy
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib as mpl


#read csv
aiueo_df = pd.read_csv("../test.csv",encoding="SHIFT-JIS")


aiueo_list = list(set(aiueo_df["カテゴリ"]))
print(aiueo_list)
for i in range(len(aiueo_list)):
    aiueo_list_t = aiueo_df.loc[aiueo_list[i]]
    aiueo_list_t.to_csv(aiueo_list[i]+".csv",encoding="SHIFT-JIS")
print("Finish!")

以下エラーが起きました。

KeyError: 'the label [縺輔＠縺吶○縺拆 is not in the [index]'

文字コードの問題だとは思うのですが、どのようにすればよろしいでしょうか。

どうぞよろしくお願い致します。

Answer 1

エラーの原因は "カテゴリ"は Indexではない為

    aiueo_list_t = aiueo_df.loc[aiueo_list[i]]

にて行を指定できずにエラーになっているかと思います。
ですので、この部分を

    aiueo_list_t = aiueo_df.loc[aiueo_df["カテゴリ"] == aiueo_list[i]]

のようにすると良いのではないでしょうか。

と書きましたが。
この処理は DataFrame.groupby() を使用したほうが楽なのではないでしょうか。

import pandas as pd

aiueo_df = pd.read_csv("../test.csv",encoding="SHIFT-JIS")

for category, df in aiueo_df.groupby("カテゴリ"):
    df.to_csv("{}.csv".format(category), encoding="SHIFT-JIS")

Comments

[essa]

こんな簡単に実現できるとは思いませんでした。
勉強になります。
ありがとうございました。