將字符串矩陣轉換為 numpy ndarray

Question

考慮如下矩陣：

+0.00000E+00 +0.00000E+00\n +0.00000E+00 +0.00000E+00\n +0.00000E+00 +0.00000E+00

最初是從文本文件中提取的，我想將其轉換為大小為 3 x 2 且帶有純 int 元素的 numpy ndarray。 我嘗試了簡單的 int function、pickle 模塊和其他一些沒有運氣的方法。

Answer 1

你可以這樣做：

string = "+0.00000E+00 +0.00000E+00\n +0.00000E+00 +0.00000E+00\n +0.00000E+00 +0.00000E+00"
array_n = re.findall(r'\S+', string)
array_n = np.array(array_n)

array_out = array_n.reshape(3,2)

這使：

array([['+0.00000E+00', '+0.00000E+00'],
       ['+0.00000E+00', '+0.00000E+00'],
       ['+0.00000E+00', '+0.00000E+00']], dtype='<U12')