Início / coding / Perguntas / 78359422
Accepted
bob.sacamento
Asked: 2024-04-21 03:09:04 +0800 CST

Tentando encontrar uma maneira pythônica de preencher parcialmente o array numpy

  • 772

Eu tenho uma matriz numpy pside formas (3.1000)

psi.__class__
Out[115]: numpy.ndarray

psi.shape
Out[116]: (3, 1000)

Quero preencher parcialmente psicom outro arrayb

b.__class__
Out[113]: numpy.ndarray

b.shape
Out[114]: (3, 500)

Posso fazer isso com um loop:

for n in range(3):
   psi[n][:500] = b[n]

Mas parece-me que deveria haver uma maneira de fazer isso de forma mais direta. Mas por exemplo

psi[:][:500] = b

falha com erro

Traceback (most recent call last):

  File "<ipython-input-120-6b23082d9d6b>", line 1, in <module>
    psi[:][:500] = b

ValueError: could not broadcast input array from shape (3,500) into shape (3,1000)

Também tenho algumas variações do tema, com resultados semelhantes. Isso parece bastante simples. Alguma ideia de como fazer isso?

python

1 respostas

  1. Best Answer

    Você pode usar:

    psi[:, :500] = b

    Ou, se n(=3) no seu loop não corresponder à primeira dimensão de b:

    idx = np.arange(3)
psi[idx, :500] = b[idx]

    Comparando as duas abordagens ( psi1com o loop, psi2como atribuição vetorial):

    np.allclose(psi1, psi2)
# True
    • 2

relate perguntas