pythonで複素数のデータを読み書き

Question

pythonで複素数のデータを読み書き

python2.7で複素数のデータを扱いたいのですが、()がついたり、読み込みがうまく行かなかったりでうまく動作しません。

該当のソースコード

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
import numpy as np

(a1) = np.genfromtxt("○○○.csv",unpack=True, dtype=np.complex128,delimiter=',')

print a1

ファイルの書き込みは
np.savetxt("○○○.csv",data , fmt='%.16f',delimiter=',')
で行いました。

dataには(1.9464469341504152e-12-3.1456638809610314e-12j), (8.78883381959785e-13-3.852325732580186e-13j), (-9.583868301245934e-13-1.873118734890551e-12j),こんな感じで複素数が入ってます。

試したこと

結果として

[-2.021478e-10+9.414360e-10j -1.836750e-11+8.232035e-10j
6.789447e-10+3.419484e-10j ... nan+0.000000e+00j
nan+0.000000e+00j -1.839600e-12+1.562000e-13j]

のように途中でnanと欠損していしまいます。csvにはしっかりと書き込まれいています。
読み込みが悪いのか、書き込みが悪いのかもよくわかりません。
よろしくお願いします。

Answer 1

numpyのバグで、虚部の符号がマイナスの場合、

出力csvファイル

 (1.9464469341504152e-12+-3.1456638809610314e-12j)
 (8.7888338195978500e-13+-3.8523257325801861e-13j)
 (-9.5838683012459342e-13+-1.8731187348905510e-12j)

のように+-というフォーマットで出力されてしまうという問題です。そのため読み込むことができません。

リンク：
python - numpy.genfromtxt give nan for complex numbers with minus signs? - Stack Overflow
savetxt complex number bug · Issue #7895 · numpy/numpy · GitHub
loadtxt does not accurately save float128 values · Issue #10436 · numpy/numpy · GitHub

悲しいことに、2016年だかに報告されて、まだ直る気配がないようです。

回避策は上のリンクの1つめのstackoverflowに載っていますが、要するに自分でコンバータ作れば良いので、やり方はそこの方法に限らず色々あると思います。

面倒なら最初からcsv出力はやめて、np.save()とnp.load()でやった方が正直良いと思います。

Comments

[konomamadehai]

返信遅れて申し訳ありません。
なるほどnumpyのバグなのですね。

np.save()とnp.loadで複素数の部分の計算はやることにしました。
ありがとうございました。