python - 데이터를 복제하지 않고 numpy 배열 반복하기

Question

이 질문은 이전에 요청되었지만 솔루션은 1D/2D 배열에서만 작동하며 좀 더 일반적인 대답이 필요합니다.

데이터를 복제하지 않고 어떻게 반복 배열을 만드나요? 메모리 사용량없이 파이썬 연산을 벡터화하는 데 도움이되므로 일반적으로 사용되는 것으로 생각됩니다.

더 구체적으로 말해서, (y, x) 배열을 만들려면 여러 번 타일링해야합니다. numpy.tile (array, (nz, 1,1))을 사용하여이 작업을 수행 할 수 있지만 메모리가 부족합니다. 내 구체적인 사례는 x = 1500, y = 2000, z = 700입니다.

Answer 1

하나의 간단한 트릭은 첫 번째 차원에서 z -long 벡터에 대한 귀하의 (x, y)을 방송 np.broadcast_arrays을 사용하는 것입니다

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 

가 this answer의 1 차원 배열에 주어진 stride_tricks 방법을 일반화하기 위해, 당신은 단지 필요 출력 배열의 각 차원에 대한 모양과 보폭을 포함하십시오.

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 
      )