python numpy將字符串轉換爲複數

Question

我有一個字符串列表，每個字符串都是一個複數。格式如下：

lr=['3.123+2.43i','0.23454-1.2134i','0.6543+0.76543i','1.12324+0.6543i'] 

我想要的是每個複數的模數列表。

Answer 1

Python的原生支持複數，但它使用Ĵ的工程約定虛數單位，而不是數學家的我的。

lr = ['3+4i', '3.123+2.43i', '0.23454-1.2134i', '0.6543+0.76543i', '1.12324+0.6543i'] 
for s in lr: 
    s = s.replace('i', 'j') 
    v = complex(s) 
    print(v, abs(v)) 

輸出

(3+4j) 5.0 
(3.123+2.43j) 3.9570227444380457 
(0.23454-1.2134j) 1.2358594465391282 
(0.6543+0.76543j) 1.006971486637035 
(1.12324+0.6543j) 1.2999140693138143 

我們可以使用列表理解創建模量清單：

moduli = [abs(complex(s.replace('i', 'j'))) for s in lr] 
print(moduli) 

輸出

[5.0, 3.9570227444380457, 1.2358594465391282, 1.006971486637035, 1.2999140693138143] 

Answer 2

下面是對'的J取代我的解決方案的變化：

In [182]: lr=['3.123+2.43i','0.23454-1.2134i','0.6543+0.76543i','1.12324+0.6543i'] 

可以使一個字符串數組：

In [183]: np.array(lr) 
Out[183]: 
array(['3.123+2.43i', '0.23454-1.2134i', '0.6543+0.76543i', 
     '1.12324+0.6543i'], 
     dtype='<U15') 

，但並不複雜 - 至少不是直接

In [184]: np.array(lr,np.complex) 
... 
TypeError: a float is required 

但有一個函數可以對數組元素執行字符串替換：

In [185]: a=np.array(lr) 
In [186]: np.char.replace(a,'i','j') 
Out[186]: 
array(['3.123+2.43j', '0.23454-1.2134j', '0.6543+0.76543j', 
     '1.12324+0.6543j'], 
     dtype='<U15') 

現在字符串的數組可被鑄造爲複雜

In [187]: np.char.replace(a,'i','j').astype(np.complex) 
Out[187]: 
array([ 3.12300+2.43j , 0.23454-1.2134j , 0.65430+0.76543j, 
     1.12324+0.6543j ])