Pythonで、数値が inf となる場合は代わりに文字列へ変換したい

Question

前提・実現したいこと

Pythonでyamlファイルを一度読み込み、別のyamlファイルとして書き出す処理を実装しようとしています。
元ファイルの中身を維持しながら別ファイルへ書き出したいのですが、Pythonで該当のyamlファイルを読み込むと一部の値がinfと判定されてしまいます。
該当の値を数値として利用することはないため、yamlファイルをloadする際、inf判定の代わりに文字列へ変換するような方法についてアドバイスをいただきたいです。

環境

Python 3.6.8
ruamel.yaml 0.17.16

発生している問題・エラーメッセージ

inf な値を含む example_in.yaml を Python で load すると、該当の値が inf に変換されてしまう。

該当のソースコード

example_in.yaml

hoge: 1e1000

example.py

from ruamel.yaml import YAML
yaml = YAML()

with open("example_in.yaml", 'rt') as f:
    data = yaml.load(f)

print(data)
# ordereddict([('hoge', inf)])

with open("example_out.yaml", 'w') as f:
    yaml.dump(data, f)

example_out.yaml

hoge: .inf

Answer 1

力技にはなりますが、inf判定される値をbaseloaderでloadした値に置き換えることで、該当の値のみを文字列にすることができ、無事回避できました。

hoge: 1e1000

from ruamel.yaml import YAML
import math

with open("example_in.yaml", 'rt') as f:
    data = YAML().load(f)

with open("example_in.yaml", 'rt') as f:
    data_baseloaded = YAML(typ="base").load(f)

for d in data:
    if data[d] == math.inf:
        data[d] = data_baseloaded[d]

with open("example_out.yaml", 'w') as f:
    YAML().dump(data, f)

hoge: '1e1000'

皆さまご助力いただきありがとうございました。

Answer 2

対策

まだ途中ですが、読み込んだ時にYAMLのFLOATでINFの時に、何行目かを出せるとこまで来ました。
あとはその行数の時に力業で値をSTRINGに差し替えればいけそうな気がします。

hoge: 1e1000
hoge1: 1e2

from ruamel.yaml import YAML
import ruamel.yaml 
yaml = YAML()

with open("example_in.yaml", 'rt') as f:
    data = yaml.load(f)

# print(data)
# ordereddict([('hoge', inf)])

for i,d in enumerate(data):
    if (type(data[d]) ==  ruamel.yaml.scalarfloat.ScalarFloat) and (str(data[d]) == "inf"):
        print("[INF DETECTED!] Line:",i, "INDEDX:",d)
    else:
        print(d,data[d])


# with open("example_out.yaml", 'w') as f:
#     yaml.dump(data, f)

駄目な投稿

yamlファイルをloadする際、inf判定の代わりに文字列へ変換する

"yaml.load" infinityと探していたら偶然見つかりました。

YAMLモジュール内の正規表現のせいでそうなっているそうです。
走らせてませんので確認はできていませんが、これで解決できそうな気がします。

Comments

[TakaiY]

いや、たぶん、違うと思います。

「1e1000」はpythonのfloatの上限(1.7976931348623157e+308)を超えてしまうので、inf (無限大)という「値」になってしまうのだと思います。

[退会済みユーザー]

そうですね、その通りと思います。しかも元の英文のタイトルは数値がSTRINGで困っているからFLOATにしてくれ的な内容なので真逆でしたね…。

[mtomsy]

ご返答ありがとうございます。
いただいた対策にアイデアを貰い、infになる値を文字列へ置換することで無事解決に至りそうです。
非常に助かりました。