Pythonのcsv
モジュールには、DictWriter
という便利なクラスがあります。このクラスを使うと、辞書型のデータをCSVファイルに書き出すことができます。
以下に、DictWriter
を使ってCSVファイルを作成する基本的なコードを示します。
import os
import csv
data = {
'Ken': {'Country': 'US', 'Job': 'Artist', 'Age': 56},
'Alan': {'Country': 'Japan', 'Job': 'Carpenter', 'Age': 27},
'Steven': {'Country': 'China', 'Job': 'Artist', 'Age': 40}
}
filename = 'test_a.csv'
file_path = os.path.join(os.getcwd(), filename)
f = open(file_path, 'w')
fieldnames = data[list(data.keys())[0]].keys() # Dict型データの1列目のKEYを取得する。
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer.writeheader() # ここでヘッダが書かれる。
for name in data:
writer.writerow(data[name])
f.close()
このコードでは、まずdata
という辞書型のデータを定義しています。次に、csv.DictWriter
を使ってCSVファイルを作成し、writeheader
メソッドでヘッダーを書き出しています。最後に、writerow
メソッドを使って各行のデータを書き出しています。
このように、Pythonのcsv.DictWriter
を使うと、辞書型のデータを簡単にCSVファイルに書き出すことができます。これは、データ分析や機械学習の分野で非常に便利な機能です。
以上、Pythonのcsv.DictWriter
の基本的な使い方について説明しました。この記事が、PythonでCSVファイルを扱う際の参考になれば幸いです。