ある列のデータが重複しているとき、違う列のデータをコンマで繋ぐ際の考え方

Question

前提・実現したいこと

現在以下のように入力されているcsv（元はExcel）があります。

大分類	品目
野菜	ニンジン
野菜	にがうり
野菜	ピーマン
果物	りんご
果物	みかん
花	チューリップ
花	桜

これを、下のようなデータに書き換えたいと思っています。

大分類	品目
野菜	ニンジン,にがうり,ピーマン
果物	りんご,みかん
花	チューリップ,桜

csvの編集といえばVBAかPythonなのかなと思い
せっかくだからとPythonの文法を調べている段階です。

発生している問題・エラーメッセージ

以下のようなフローを考えました。（用語の不正確さはご容赦ください）

1. 元となる配列（今回はcsv）を準備する
2. 元となる配列をforで回す
3. 何らかの変数を準備して、初出の大分類をキーにした空配列を作る
4. 既出の大分類が出現したら品目の要素を↑の配列に入れてから削除する
5. もう一度forを回して、↑の情報を上書きしていく

もっとすっきり書けないものかと思うのですが、どうにも良い方法が思いつきません。
今回の場合、どのような考え方で処理を行っていくのが正解でしょうか。

該当のソースコード

考え方の時点で？となったので、コードはcsvをprintするところで止まっています。

import pandas as pd

df = pd.read_csv('test.csv').values.tolist()

for key,value in enumerate(df):
    print("{0} = {1[0]}".format(key,value))

補足情報（FW/ツールのバージョンなど）

Python：3.6

追記

ひとまず書けるところまで書いてみました。（当初の考え方とは変わってしまいましたが…）
思っているような結果が出ることは確認済みです。
メソッドの書き方がこれで本当に正しいのかは自信がありません。

class Sample:

    def __init__(self, readCsvFile, writeCsvFile):
        self.readCsvFile = readCsvFile
        self.writeCsvFile = writeCsvFile

    def read(self):
        import pandas as pd
        return pd.read_csv(self.readCsvFile).values.tolist()

    def edit(self, df):
        from itertools import groupby

        # 大分類初出リスト
        firstAppearanceList = []
        # 項目リスト
        itemsList = []

        for key, group in groupby(df, key=lambda x: x[0]):
            # 大分類をまたいだ項目の重複は可とする（野菜のスイカと果物のスイカは別扱い）
            seen = []
            firstAppearanceList.append(key)
            itemsList.append(','.join([x[1] for x in list(group) if x[1] not in seen and not seen.append(x[1])]))

        return list(zip(firstAppearanceList, itemsList))

    def write(self, list):
        import csv
        f = open(self.writeCsvFile, 'w')
        writer = csv.writer(f, lineterminator='\n')
        writer.writerows([x for x in list])

def done(readCsvFile, writeCsvFile):
    sample = Sample(readCsvFile, writeCsvFile)
    df = sample.read()
    list = sample.edit(df)
    sample.write(list)

print('インポート（加工）したいcsvのファイル名を入力してください。')
readCsvFile = input('>> ') or 'test.csv'
print('エクスポートする際のcsvのファイル名を入力してください。')
writeCsvFile = input('>> ') or 'tmp.csv'
done(readCsvFile, writeCsvFile)

Comments

[mather]

「もっとすっきり書けないものかと思うのですが、どうにも良い方法が思いつきません。」とありますが、最初は愚直でもいいので自分の力で動くものを書いてください。アドバイスを受けるのはそれからです。

[chibi144]

ご依頼ありがとうございます。ひとまず上の考え方で書けるところまで書いてみて（書ききれるかも怪しいですが）、その分を追記します。

[chibi144]

追記しました。今までに頂いているご回答はまだ全く見れていないので、これからじっくり読ませていただきます。

Answer 1

import pandas as pd

df = pd.read_csv('test.csv', encoding='Shift_JIS') # エクセルで作成したcsvファイルを読み込み
df['品目'] = df['品目']+',' # sum()適用時にカンマ入れれないためここで入れる
grouped = df.groupby(['大分類'])
df2 = grouped.sum()
for i in range(len(df2.index)): # 末尾の不要なカンマを取り除く
    df2['品目'][i] = df2['品目'][i][:-1]
print(df2) # Jupyterならdf2のみで打ち出すと見やすいです

やや強引ですが、こんな感じでどうでしょうか？

Answer 2

こういう事でしょうか？

import pandas as pd
import io

data = """
大分類,品目
野菜,ニンジン
野菜,にがうり
野菜,ピーマン
果物,りんご
果物,みかん
花,チューリップ
花,桜"""

df = pd.read_csv(io.StringIO(data))
df = df.groupby('大分類')['品目'].apply(list)
print(df)
#大分類
#果物            [りんご, みかん]
#花            [チューリップ, 桜]
#野菜    [ニンジン, にがうり, ピーマン]
#Name: 品目, dtype: object

Answer 3

行によって列数が違うcsvにするということでしたら、pandasはあまり向いていない気がします。
csvモジュールを使って愚直に書いてみました。

import io
import csv
from collections import defaultdict

data = """大分類,品目
野菜,ニンジン
野菜,にがうり
野菜,ピーマン
果物,りんご
果物,みかん
花,チューリップ
花,桜"""

# データを読み込んでdefaultdictに入れる
input_f = io.StringIO(data)  # io.StringIOの代わりに入力のファイルオブジェクトを置けます
reader = csv.reader(input_f)

header = next(reader)
d = defaultdict(list)
for cls, item in reader:
    d[cls].append(item)
input_f.close()


# csvに書き出す
output_f = io.StringIO()    # io.StringIOの代わりに出力のファイルオブジェクトを置けます
writer = csv.writer(output_f)
writer.writerow(header)
for cls, items in d.items():
    writer.writerow([cls, *items])
print(out.getvalue())  # io.StringIOの中身を確認するためのコードで、実際は不要です
out.close()

""" =>
大分類,品目
果物,りんご,みかん
野菜,ニンジン,にがうり,ピーマン
花,チューリップ,桜
"""