類似アイテムのリストに基づいてファイルを作成して書き込む

Question

237行のトランザクションを含むCSVファイルがありますが、すでに正しい順序でトランザクションを分類してソートし、新しいCSVファイルを作成しました。口座番号でしかし、私はさらに一歩進んで、その口座番号の取引すべてで口座番号に基づいて複数のCSVファイルを作成したいと思います。

アカウント番号に基づいて適切なCSVファイルに各行を追加するだけですが、これを動的に実行したいので、各変数を作成する必要はありません常に未知数の口座番号。

Account #, Date, Dollar Amount (stored as heading) 
001, 1/1/11, $25 
001, 1/1/11, $20 
002, 1/1/11, $15 
003, 1/4/11, $19 

このデータでは、アカウント番号に基づいて3つの別々のCSVファイルを作成したいと考えています。私は誰かがコードを書くのを探しているわけではありませんが、Pythonにはこのタイプのプロセスが関与していることに気づいていないモジュールがいくつかあります。誰かが私を正しい方向に向けることができたら、私はそれを高く評価します。

ありがとうございました

Answer 1

これはすばやく効率的ではないソリューションです。

from sets import Set 
import csv 

source = list(csv.reader(open("account_file.csv"))) 

accounts = Set(line[0] for line in source) 

for account in accounts: 
    out = open(account+".csv","w") 
    out.write("\n".join(",".join(x) for x in source if x[0] == account)) 
    out.close() 

Answer 2

このようなモジュールはありません。私は、アカウント操作ごとに別々のリストを作成し、keyがアカウント番号である辞書に入れます。入力ファイル全体を反復処理した後、各リストを保存して.csvファイルに分割することができます。

入力ファイルがすでにソートされている場合は、行を出力ファイルにコピーしてから、アカウントを変更すると出力ファイルを閉じ、新しいアカウントの次の出力ファイルを開くことができます。

Answer 3

CSVインターフェースの必要がありません、私はこれを好む：

from collections import defaultdict 
from os.path import getsize 

with open('accounts.csv') as f: 

    first_line = f.readline() 

    d = defaultdict(list) 
    for line in f: 
     d[line.split(',')[0]].append(line) 

    for account,lines in d.iteritems(): 
     with open(account+'.csv','a') as f: 
      if getsize(account+'.csv')==0: 
       f.write(first_line) 
      f.write(''.join(lines)) 

ファイルを、このような作業に一度だけ