Füllen Sie die Numpy-Matrix/-Maske vorwärts mit Werten basierend auf der Bedingung ⇐ Python

[Guest]

Ich habe die folgende Matrix

Code: Select all

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)

Wie fülle ich alle Zeilen spaltenweise, nachdem eine Spalte wahr ist?
Meine gewünschte Ausgabe:

Code: Select all

    [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]