Início / coding / Perguntas / 79298424
Accepted
Paul Jurczak
Asked: 2024-12-21 05:22:12 +0800 CST

Existe alguma mágica no NumPy para gerar esse array?

  • 772

Tenho um array ade índices crescentes, por exemplo [2 5 9 10], , que indica posições de mudança de valor. Supondo que os valores de saída sejam 0e 1, quero obter o array b:

[0 0 1 1 1 0 0 0 0 1 0]

Existe alguma mágica no NumPy para se atransformar b?

python

3 respostas

  1. Best Answer

    Um caminho entre muitos outros

    a=np.array([2,5,9,10])
x=np.zeros((a.max()+1,), dtype=np.uint8)
x[a]=1
b=x.cumsum()%2

    Alguma explicação (mas acho que o código, neste caso raro, é sua própria explicação, já que é bem fácil, uma vez que você o vê) x (after x[a]=1) contém 1 em cada posição dada em a. Então x.cumsum()contém um valor que incrementa para cada um desses valores: 0 para o 2 primeiro, então 1 para o 3 depois, então 2, então 3, então 4... Então x.cumsum()%2alterna entre 1 e 0.

    Note que eu uso np.uint8type porque sou econômico, e não consigo deixar de pensar "por que eu deveria pagar 32 bits quando 8 são o suficiente para um array de tamanho 11". Mas na realidade, já que 256 é par, não importaria mesmo se ativesse bilhões de valores. Apenas x.cumsum()voltaria de 255 para 0 por causa do estouro. E então x.cumsum()%2teria o mesmo valor.

    • 11

  2. Provavelmente você pode tentar np.diff+ np.repeatcomo abaixo

    d = np.diff([0] + a.tolist() + [max(a).tolist()+1])
np.repeat((np.arange(d.size)) % 2, d)

    o que dá

    array([0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0])
    • 3

  3. Se S = [2 5 9 10] e res o resultado final, observando que o valor é 0 em todos os lugares, exceto dentro das fatias S[0]:S[1], S[2]:S[3], ... onde é 1:

    res = np.zeros(max(S)+1)
idx = [slice(S[i], S[i+1] if i+1!=len(S) else len(res)) 
       for i in np.arange(len(S)) if not i%2]
res[np.r_[*idx]] = 1

[0. 0. 1. 1. 1. 0. 0. 0. 0. 1. 0.]
    • 1

relate perguntas