将字符串矩阵转换为 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')