python - ripetizione di array numpy senza replica dei dati

Question

Questa domanda è stata posta prima, ma la soluzione funziona solo per gli array 1D/2D e ho bisogno di una risposta più generale.

Come si crea un array ricorrente senza la replica dei dati? Questo mi sembra una cosa di uso generale, poiché aiuterebbe a vettorializzare le operazioni Python senza il colpo di memoria.

In particolare, ho una matrice (y, x), che desidero affiancare più volte per creare una matrice (z, y, x). Posso farlo con numpy.tile (array, (nz, 1,1)), ma a corto di memoria. Il mio caso specifico ha x = 1500, y = 2000, z = 700.

Answer 1

Un semplice trucco è quello di utilizzare np.broadcast_arrays per trasmettere la tua (x, y) contro un z vettore -lungo nella prima dimensione:

import numpy as np 

M = np.arange(1500*2000).reshape(1500, 2000) 
z = np.zeros(700) 

# broadcasting over the first dimension 
_, M_broadcast = np.broadcast_arrays(z[:, None, None], M[None, ...]) 

print M_broadcast.shape, M_broadcast.flags.owndata 
# (700, 1500, 2000), False 

di generalizzare il metodo stride_tricks data per un array 1D in this answer, non vi resta che includi la forma e la lunghezza della falcata per ogni dimensione della matrice di output:

M_strided = np.lib.stride_tricks.as_strided(
       M,        # input array 
       (700, M.shape[0], M.shape[1]), # output dimensions 
       (0, M.strides[0], M.strides[1]) # stride length in bytes 
      )