如何从文本文件中导入与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.]])