Início / coding / Perguntas / 78978186
Accepted
Quercus Robur
Asked: 2024-09-12 20:38:10 +0800 CST

Maneira correta de encontrar dimensão após transmissão em numpy

  • 772

Suponha que eu receba alguns arrays numpy, digamos a, be c, que são assumidos como transmissíveis. Existe um padrão ou, de outra forma, uma maneira elegante de encontrar o formato do array após a transmissão? Claro, algo como (a+b+c).shapefuncionaria, mas é muito ineficiente se eu estiver interessado apenas no formato do resultado.

python

2 respostas

  1. Best Answer

    Você pode usar broadcast_shapes. Curtir isso

    result_shape = np.broadcast_shapes(a.shape, b.shape, c.shape)

    Aqui está a página de documentação: https://numpy.org/doc/stable/reference/generated/numpy.broadcast_shapes.html

    Boa sorte!

    • 2

  2. Você pode usar np.broadcast(a, b, c).shape para obter o formato sem realmente executar o cálculo. Isso é eficiente e evita criar o array completo.

    • 1

relate perguntas