b = [sum(x for x in a) for i in range] n - ширина матрицы (ну или длина вложенных массивов, если так проще) каждый элемент b содержит сумму элементов соответствующего столбца из матрицы a. если какой-то элемент b равен высоте матрицы a, значит этот столбец состоит из единиц.