Início / coding / Perguntas / 79336023
Accepted
HJA24
Asked: 2025-01-07 20:36:21 +0800 CST

Preenchimento direto da matriz/máscara Numpy com valores baseados na condição

  • 772

Eu tenho a seguinte matriz

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)

Como preencho todas as linhas coluna por coluna depois que uma coluna é True?

Minha saída desejada:

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

3 respostas

  1. Best Answer

    Você pode usar logical_orcombinado com accumulate:

    np.logical_or.accumulate(A, axis=1)

    Saída:

    array([[False, False, False, False,  True,  True,  True],
       [False, False, False, False, False, False,  True],
       [ True,  True,  True,  True,  True,  True,  True],
       [False, False, False, False, False, False, False]])

    Se você quiser números inteiros, vá com maximum:

    np.maximum.accumulate(A.astype(int), axis=1)

array([[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]])
    • 6

  2. você pode fazer um cumsum .

    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)

B = np.cumsum(A, axis=1).astype(bool) # makes copy
np.cumsum(A, axis=1, out=A) # in-place no copy


    [[False False False False  True  True  True]
 [False False False False False False  True]
 [ True  True  True  True  True  True  True]
 [False False False False False False False]]

    Edição: a resposta do mozway é mais rápida, use-a.

    • 5

  3. Aqui está uma abordagem depandas.DataFrame.cummax

    import pandas as pd

pd.DataFrame(A).cummax(axis = 1).to_numpy()

    o que dá

    array([[False, False, False, False,  True,  True,  True],
       [False, False, False, False, False, False,  True],
       [ True,  True,  True,  True,  True,  True,  True],
       [False, False, False, False, False, False, False]])
    • 0

relate perguntas