Посчитайте число нулевых столбцов в матрице

Question

Единственное, что не получается - определить вводиться ли одна строка или один столбец, потому что метод shape дает только одно значение в кортеже

import numpy as np
import sys
matrix = np.loadtxt(sys.stdin)
s = 0
a = np.array(matrix)
for i in range(0,a.shape[1]):
    matrix = np.array(a[:,i],int)
    if np.all(matrix == 0):
        s += 1
print(s)

Answer 1

Зачем так сложно? Во-первых в matrix у вас уже ndarray. Во-вторых действуйте векторно, иначе зачем вам вообще Numpy? Цикл не нужен:

import numpy as np
import sys

matrix = np.loadtxt(sys.stdin)
print((matrix==0).all(axis=0).sum())

БЛОГ НА HUSL

Посчитайте число нулевых столбцов в матрице

Ответы (1 шт):