Максимальная сумма столбца и строки матрицы в питоне

Question

Я беру ввод матрицы

import numpy as np
l = np.array([input().split() for _ in range(3)], dtype=np.int)

1 2 3
4 5 6
7 8 9

Теперь я хочу получить на дисплее наибольшую сумму, которая может быть в столбце, а также в строке также

Например, в этой строке 3имеет максимальную сумму 24

, поэтому мой вывод будет: строка 3 24

Answer 1

Рабочий пример:

import numpy as np

x = np.array([[1,2,3],[4,5,6],[7,8,9]]);

rowSum = np.sum(x, axis=1)
colSum = np.sum(x, axis=0)

print("row {} {}".format(np.argmax(rowSum)+1, np.max(rowSum)))
print("col {} {}".format(np.argmax(colSum)+1, np.max(colSum)))

# output:
# row 3 24
# col 3 18

См.

• https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.argmax.html

• https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.sum.html