Saves and loads python dictionaries to json

save_load_json_dict.py

import json
import numpy as np
from json import JSONEncoder

def open_data_file(filename):
    # Load data from json file
    with open(filename, 'r') as fp:
        json_dict = json.load(fp)
    return json_dict
        
class NumpyArrayEncoder(JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.ndarray):
            return obj.tolist()
        return JSONEncoder.default(self, obj)

def save_data_file(data_dict, filename, indent=4):
    """ Saves dictionary to filename"""
    with open(filename, 'w') as fp:
        json.dump(data_dict, fp, cls=NumpyArrayEncoder, indent=indent)