Как импортировать массив из текстового файла в формате 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

Просто используйте repr при печати массива numpy, который сохранит все запятые нетронутыми, обычно 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.]])