日々精進

新しく学んだことを書き留めていきます

OrderedDictをyamlに出力する

yaml.dump(ordered_dict)を実行するだけだと、Pythonのクラス名も出力されてしまうので以下のようにすると通常のyaml形式で出力される。

        def represent_odict(dumper, instance):
            return dumper.represent_mapping('tag:yaml.org,2002:map', instance.items())
        yaml.add_representer(OrderedDict, represent_odict)
        return yaml.dump(od, default_flow_style=False)

参考:

qiita.com