Python で json ファイルを扱う方法

json ファイルを辞書型で読み込む方法と書き込む方法について解説します．

書き込み

コード

import json
import pprint

# dict の作成
params = {}
params["epoch"] = 20
params["lr"] = 1e-5
params["batch_size"] = 64
params["sample_list"] = [i for i in range(10)]

pprint.pprint(params)
# {'batch_size': 64,
#  'epoch': 20,
#  'lr': 1e-05,
#  'sample_list': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]}

# params という dict を ./kwags.json  に書き込む
dict_name = './kwags.json'
with open(dict_name, 'w') as f:
    json.dump(params, f, indent=4)

作成される json ファイル

{
    "epoch": 20,
    "lr": 1e-05,
    "batch_size": 64,
    "sample_list": [
        0,
        1,
        2,
        3,
        4,
        5,
        6,
        7,
        8,
        9
    ]
}

読み込み

読み込む json ファイル

上で作成した json ファイルを読み込む

{
    "epoch": 20,
    "lr": 1e-05,
    "batch_size": 64,
    "sample_list": [
        0,
        1,
        2,
        3,
        4,
        5,
        6,
        7,
        8,
        9
    ]
}

コード

import json
import pprint

# ./kwags.json  から params という dict で読み込む
param_dir = "./kwags.json"
tf = open(param_dir, "r")
params = json.load(tf)

pprint.pprint(params)
# {'batch_size': 64,
#  'epoch': 20,
#  'lr': 1e-05,
#  'sample_list': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]}

以上参考になれば嬉しいです．