Início / coding / Perguntas / 79224958
Accepted
Ben
Asked: 2024-11-26 08:30:40 +0800 CST

Formar lista elemento a elemento a partir de escalar e matriz

  • 772

numpyEu tenho um escalar zero-dimensional e uma matriz sbidimensional . Eu quero formar uma matriz de vetores em que todos os elementos de são pareados com como no exemplo a seguir:numpymms

import numpy as np

s = np.asarray(5)

m = np.asarray([[1,2],[3,4]])

# Result should be as follows

array([[[5, 1],
        [5, 2]],

       [[5, 3],
        [5, 4]]])

Em outras palavras, quero vetorizar a operação np.asarray([s, m])elemento a elemento no nível mais baixo de m. Existe uma maneira óbvia de fazer isso para qualquer array multidimensional mdentro de numpy?

Tenho certeza de que isso está em algum lugar, mas tenho dificuldade em expressar em palavras e não consigo encontrar. Se você puder encontrar, sinta-se à vontade para me redirecionar para lá.

numpy

2 respostas

  1. Best Answer

    Uma solução possível, que usa broadcast_toe stackfunções para combinar dois arrays, se m, em um único array ao longo de um novo eixo. Os passos são:

    • Primeiro, np.broadcast_to(s, m.shape)expande o formato do array spara corresponder ao do array msem copiar dados.

    • Em seguida, np.stack([np.broadcast_to(s, m.shape), m], axis=-1)junta-se ao transmitido se mao longo de um novo último eixo

    np.stack([np.broadcast_to(s, m.shape), m], axis=-1)

    Saída:

    array([[[5, 1],
        [5, 2]],

       [[5, 3],
        [5, 4]]])
    • 2

  2. O stackestá bom. Mas como sprecisa ser repetido, uma boa alternativa é criar um array zero do formato desejado, e permitir broadcastingpreencher as colunas:

    In [2]: s = np.asarray(5)
   ...: 
   ...: m = np.asarray([[1,2],[3,4]])

In [3]: res = np.zeros((m.shape)+(2,)); res
Out[3]: 
array([[[0., 0.],
        [0., 0.]],

       [[0., 0.],
        [0., 0.]]])

In [4]: res[:,:,0] = s; res[:,:,1] = m

In [5]: res
Out[5]: 
array([[[5., 1.],
        [5., 2.]],

       [[5., 3.],
        [5., 4.]]])
    • 0

relate perguntas