Печать двумерного массива в python
Мне нужно напечатать этот код python в массиве 5x5, массив должен выглядеть так:
0 1 4 (infinity) 3
1 0 2 (infinity) 4
4 2 0 1 5
(inf)(inf) 1 0 3
3 4 5 3 0
может кто-нибудь помочь мне распечатать эту таблицу? используя индексы.
for k in range(n):
for i in range(n):
for j in range(n):
if A[i][k]+A[k][j]<A[i][j]:
A[i][j]=A[i][k]+A[k][j]
Ответы
Ответ 1
Комбинация понимания списка и str
присоединяется может выполнить задание:
inf = float('inf')
A = [[0,1,4,inf,3],
[1,0,2,inf,4],
[4,2,0,1,5],
[inf,inf,1,0,3],
[3,4,5,3,0]]
print('\n'.join([''.join(['{:4}'.format(item) for item in row])
for row in A]))
дает
0 1 4 inf 3
1 0 2 inf 4
4 2 0 1 5
inf inf 1 0 3
3 4 5 3 0
Использование for-loops с индексами обычно можно избежать в Python и не считается "Pythonic", потому что оно менее читаемо, чем его кузен Pythonic (см. ниже). Однако вы можете сделать это:
for i in range(n):
for j in range(n):
print '{:4}'.format(A[i][j]),
print
Чем больше кузенец-питонов будет:
for row in A:
for val in row:
print '{:4}'.format(val),
print
Однако это использует 30 операторов печати, тогда как мой первоначальный ответ использует только один.
Ответ 2
Всегда есть простой способ.
import numpy as np
print(np.matrix(A))
Ответ 3
Я использовал numpy для создания массива, но список массивов списков должен работать аналогичным образом.
import numpy as np
def printArray(args):
print "\t".join(args)
n = 10
Array = np.zeros(shape=(n,n)).astype('int')
for row in Array:
printArray([str(x) for x in row])
Если вы хотите напечатать только определенные индексы:
import numpy as np
def printArray(args):
print "\t".join(args)
n = 10
Array = np.zeros(shape=(n,n)).astype('int')
i_indices = [1,2,3]
j_indices = [2,3,4]
for i in i_indices:printArray([str(Array[i][j]) for j in j_indices])
Ответ 4
с использованием индексов, для циклов и форматирования:
import numpy as np
def printMatrix(a):
print "Matrix["+("%d" %a.shape[0])+"]["+("%d" %a.shape[1])+"]"
rows = a.shape[0]
cols = a.shape[1]
for i in range(0,rows):
for j in range(0,cols):
print "%6.f" %a[i,j],
print
print
def printMatrixE(a):
print "Matrix["+("%d" %a.shape[0])+"]["+("%d" %a.shape[1])+"]"
rows = a.shape[0]
cols = a.shape[1]
for i in range(0,rows):
for j in range(0,cols):
print("%6.3f" %a[i,j]),
print
print
inf = float('inf')
A = np.array( [[0,1.,4.,inf,3],
[1,0,2,inf,4],
[4,2,0,1,5],
[inf,inf,1,0,3],
[3,4,5,3,0]])
printMatrix(A)
printMatrixE(A)
который дает выход:
Matrix[5][5]
0 1 4 inf 3
1 0 2 inf 4
4 2 0 1 5
inf inf 1 0 3
3 4 5 3 0
Matrix[5][5]
0.000 1.000 4.000 inf 3.000
1.000 0.000 2.000 inf 4.000
4.000 2.000 0.000 1.000 5.000
inf inf 1.000 0.000 3.000
3.000 4.000 5.000 3.000 0.000
Ответ 5
print(mat.__str__())
где мат является переменной, относящейся к вашему объекту-матрице
Ответ 6
В дополнение к простому ответу на печать вы можете настроить вывод на печать с помощью функции numpy.set_printoptions.
Предпосылки:
>>> import numpy as np
>>> inf = np.float('inf')
>>> A = np.array([[0,1,4,inf,3],[1,0,2,inf,4],[4,2,0,1,5],[inf,inf,1,0,3],[3,4,5,3,0]])
Следующая опция:
>>> np.set_printoptions(infstr="(infinity)")
Результаты в:
>>> print(A)
[[ 0. 1. 4. (infinity) 3.]
[ 1. 0. 2. (infinity) 4.]
[ 4. 2. 0. 1. 5.]
[(infinity) (infinity) 1. 0. 3.]
[ 3. 4. 5. 3. 0.]]
Следующая опция:
>>> np.set_printoptions(formatter={'float': "\t{: 0.0f}\t".format})
Результаты в:
>>> print(A)
[[ 0 1 4 inf 3 ]
[ 1 0 2 inf 4 ]
[ 4 2 0 1 5 ]
[ inf inf 1 0 3 ]
[ 3 4 5 3 0 ]]
Если вы просто хотите получить конкретный вывод строки для определенного массива, функция numpy.array2string также доступна.