Python: инициализировать массивы numpy в массиве нулей

Question

В Python я пытаюсь инициализировать двухэлементные массивы нулей в пределах размера N на N. Код, который я использую, работает, но я ищу что-то более эффективное и элегантное:

array1 = np.empty((N,N), dtype=object)
for i in range(N):
    for j in range(N):
        array1[i,j] = np.zeros(2, dtype=np.int)

Спасибо за помощь

Answer 1

Насколько я понимаю, вам, вероятно, следует использовать трехмерный массив:

import numpy as np
array1 = np.empty((N,N,2), dtype=object)

, который возвращает массив из N строк, N столбцов и 2 глубин. Если вы хотите передать массив (NxN), скажем, первую глубину, просто используйте:

tmp = np.ones(N,N) #for instance
array1(:,:,0) = tmp