Вперед заполнить матрицу/маску Numpy значениями на основе условия ⇐ Python

[Anonymous]

У меня есть следующая матрица
import numpy as np

A = np.array([
[0, 0, 0, 0, 1, 0, 1],
[0, 0, 0, 0, 0, 0, 1],
[1, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0]
]).astype(bool)

Как заполнить все строки по столбцам после того, как столбец имеет значение True?
Мой желаемый результат:
[0, 0, 0, 0, 1, 1, 1],
[0, 0, 0, 0, 0, 0, 1],
[1, 1, 1, 1, 1, 1, 1],
[0, 0, 0, 0, 0, 0, 0]


Подробнее здесь: https://stackoverflow.com/questions/793 ... -condition