複雑な配列のスマートプリント

Question

複雑な浮動小数点行列が本当に実数または虚数のどちらかで満たされているのは、よくある状況です。時にはそれらも整数である（しかし、中間の数はそうではありません。そして、とにかく複雑な整数型はありません）。おそらく、すべて手作業でフォーマットすることなく、スマートなソリューションがありますか？

例：（必要とされているものにできるだけ近い）の組み合わせ

numpy.around(numpy.real_if_close(numpy.array([[1.0j,1.0],[0.0,1.0j]],complex))) 

私は

array([[ 0.+1.j, -1.+0.j], 
     [ 0.+0.j, 0.+1.j]]) 

所望の出力を得るには、私はないです

array([[ 1j, -1 ], 
     [ 0 , 1j]]) 

Answer 1

>>> import numpy 
>>> a = numpy.around(numpy.real_if_close(numpy.array([[1.0j,0.0],[0.0,1.0j]],complex))) 
>>> a 
array([[ 0.+1.j, 0.+0.j], 
     [ 0.+0.j, 0.+1.j]]) 
>>> [[c if c.imag else c.real for c in b] for b in a] 
[[1j, 0.0], [0.0, 1j]] 

あるとそれを行うための方法で構築されていることを確認します。

編集：numpyの配列をサブクラス化することをお勧めします。これは、最も簡単な方法です。そうでなければあなただけの配列からすべてを取得し、リストに入れて、しかし、あなたが望むようにそれが0を印刷していないキャストすることができ、それ（それが印刷さ0J）

>>> [[c for c in b] for b in a] 
[[1j, 0j], [0j, 1j]]