id, height, width, weightの辞書リストから、id, height, widthだけのcsvをつくっています。
import csv
outpath = '/path/to/csv'
paramnames = ['id','height','width']
rows = [{'id':0,'height':10,'width':5,'weight':3},{'id':1,'height':3,'width':7,'weight':9}]
header = dict([(val,val)for val in paramnames])
with open(outpath, mode='w') as f:
rows.insert(0,header)
writer = csv.DictWriter(f, paramnames, extrasaction='ignore')
writer.writerows(rows)
paramnamesにid, height, widthを指定しているので、
| id | height | width |
|---|---|---|
| 0 | 10 | 5 |
| 1 | 3 | 7 |
辞書のサブセットをcsv化する方法は、
csv.DictWriterでextrasaction='ignore'にすると全パラメータを使わなくてもエラーをはきません。(ignoreしてますからね・・・)
DictWriterでヘッダを追加する方法はこちらを参考にしました。
0 件のコメント:
コメントを投稿