如何从文本文件中以与numpy数组相同的格式导入数组？

Question

我有一个文本文件，其中包含一个多维数组，例如：

1  2  3
1  2  3
1  2  3
1  2  3.

当我使用np.loadtxt导入文本文件时。我得到的数组格式为：

[[1.0  2.0  3.0]
 [1.0  2.0  3.0]
 [1.0  2.0  3.0]
 [1.0  2.0  3.0]]

但是我要使用以下格式：

[[1.0, 2.0, 3.0],
 [1.0, 2.0, 3.0],
 [1.0, 2.0, 3.0],
 [1.0, 2.0, 3.0]]

我该怎么办？

Answer 1

在打印numpy数组时只需使用repr，这将使所有逗号保持完整，常规print会删除所有逗号！或者您可以使用https://docs.python.org/3/library/pprint.html#pprint.pprint

import numpy as np
import pprint
arr = np.loadtxt('file.json')
print(arr)
#[[1. 2. 3.]
# [1. 2. 3.]
# [1. 2. 3.]
# [1. 2. 3.]]
print(repr(arr))
#array([[1., 2., 3.],
#       [1., 2., 3.],
#       [1., 2., 3.],
#       [1., 2., 3.]])
pprint.pprint(arr)
#array([[1., 2., 3.],
#       [1., 2., 3.],
#       [1., 2., 3.],
#       [1., 2., 3.]])