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

chainer_log_to_csv.py

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

import argparse
import csv
import json

def convert(input_name,output_name):
	log_list = json.load(open(input_name))
	
	csv_writer = csv.writer(open(output_name, 'w'))
	
	header_list = list(sorted(max([one.keys() for one in log_list],key=len)))
	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])

if __name__=="__main__":
	parser = argparse.ArgumentParser()
	parser.add_argument('input')
	parser.add_argument('output')
	args = parser.parse_args()

	convert(args.input,args.output)