ある一つの列で同じ値を持つもの同士を1行にまとめるコード

Question

現在以下のような内容のcsvファイル「hotelResister.csv」があり、

"部屋","名前","苗字"
"Office","ダニング","スミス"
"Office","ローザ","フォックス"
"215","カイル","ハイド"
"219","メリッサ","ウッドワード"
"219","ケビン","ウッドワード"
"213","ジェフ","エンゼル"

このファイルを

・"部屋"列に同じ値をもつ行を、元のファイルでの順番を維持したまま1行にまとめる
・1行にまとめられたもの同士はスペースで区切られる
・まとめたあとは"部屋"列の値は削除

といった決まりに従いPythonで加工して、

"ダニング","スミス" "ローザ","フォックス"
"カイル","ハイド"
"メリッサ","ウッドワード" "ケビン","ウッドワード"
"ジェフ","エンゼル"

という内容の「heyawari.txt」というファイルを作りたいのですが、
これはどのような処理を書けば実現できるでしょうか。
お手数ですがご教示お願いします。

Answer 1

pandas にて実現するとこんな感じになります。
やっていることは

• "名前"と"苗字" を結合して "フルネーム" Column を作成
• DataFrame.groupby("部屋") にて同じ部屋の "フルネームを結合し、unstack() にて行に展開
• DataFrame.to_csv() にて出力 (indexとheaderは出力しない、セパレータはスペースを使用)

となります

import pandas as pd
import io

data = '''
"部屋","名前","苗字"
"Office","ダニング","スミス"
"Office","ローザ","フォックス"
"215","カイル","ハイド"
"219","メリッサ","ウッドワード"
"219","ケビン","ウッドワード"
"213","ジェフ","エンゼル"
'''

df = pd.read_csv(io.StringIO(data))
df['フルネーム'] = '"' + df["名前"] + '","' + df["苗字"] + '"'
ret = df.groupby('部屋')['フルネーム'].apply(lambda d: d.reset_index(drop=True)).unstack()
ret.to_csv('heyawari.txt', index=None, header=None, sep=' ', quotechar="'")

# "ジェフ","エンゼル"
# "カイル","ハイド"
# "メリッサ","ウッドワード" "ケビン","ウッドワード"
# "ダニング","スミス" "ローザ","フォックス"

Comments

[Terratale]

回答ありがとうございました。
参考にさせていただきました！
あと、StringIOって初めて知りました。面白いですね。

Answer 2

以下のようにOrderedDictに部屋列をキーとし、リスト値として名前,苗字を追加していけばよいです。
ファイル入出力処理は簡単のため省略してます。

csv = [["Office","ダニング","スミス"],
    ["Office","ローザ","フォックス"],
    ["215","カイル","ハイド"],
    ["219","メリッサ","ウッドワード"],
    ["219","ケビン","ウッドワード"],
    ["213","ジェフ","エンゼル"]]

from collections import OrderedDict
d = OrderedDict()

# 日時をキーとして、それ以降の列値をリストとして保持
for row in csv:
    if row[0] in d:
        d[row[0]] += row[1:]
    else:
        d[row[0]] = row[1:]

# 結果
for key,val in d.items():
    print(val)
"""
['ダニング', 'スミス', 'ローザ', 'フォックス']
['カイル', 'ハイド']
['メリッサ', 'ウッドワード', 'ケビン', 'ウッドワード']
['ジェフ', 'エンゼル']
"""

Comments

[Terratale]

回答ありがとうございました！
OrderDictなんてものがあるんですね。勉強になりました！
列数が変わってもプログラムを書き換えなくて良いのが気に入ったのでベストアンサーにさせていただきました！

Answer 3

Pythonなど不要。そう、シェル芸ならね！

$ cat hotelResister.csv|awk -F, 'NR>1{k=$1;v=$2","$3;if(b[k]){b[k]=b[k]" "v}else{a[i++]=k;b[k]=v}}END{for(i in a){print b[a[i]]}}' > heyawari.txt
$ cat heyawari.txt
"ダニング","スミス" "ローザ","フォックス"
"カイル","ハイド"
"メリッサ","ウッドワード" "ケビン","ウッドワード"
"ジェフ","エンゼル"

Comments

[Terratale]

回答ありがとうございました。
シェルでも出来るんですね！
参考になりました。