chainerのjsonログファイルをcsv形式に変換する

chainer_log_to_csv.py

"""
python chainer_log_to_csv.py [input path] [output path]
"""

import argparse
import csv
import json

parser = argparse.ArgumentParser()
parser.add_argument('input')
parser.add_argument('output')
args = parser.parse_args()

log_list = json.load(open(args.input))

csv_writer = csv.writer(open(args.output, 'w'))

header_list = list(sorted(log_list[0].keys()))
csv_writer.writerow(header_list)

for log in log_list:
    csv_writer.writerow([log[header] if header in log else None for header in header_list])

pandasのread_jsonでやろうとして失敗していたところ、参考になりました。
対話環境からimportで実行できるようにソースコードを書き換えて使用させていただいています。