Создание многомерного массива, содержащего единичную матрицу ⇐ Python

[Anonymous]

Предположим, у меня есть массив Numpy с индексами n, где первые n-2 представляют некоторые индексы счета, а последние два индекса представляют собой квадратную матрицу MxM. Я хочу инициализировать эту структуру, чтобы она содержала копии единичной матрицы.
Пример (здесь N=3, M=2):

Код: Выделить всё

A = numpy.zeros((3,2,2))
for k in range(3):
A[k,:,:] = numpy.eye(2)

Другой пример (здесь N=4, M=5):

Код: Выделить всё

B = numpy.zeros((3,4,5,5))
for k1 in range(3):
for k2 in range(4):
B[k1,k2,:,:] = numpy.eye(5)

Есть ли более компактный способ сделать это вместо (вложенных) циклов?

Подробнее здесь: https://stackoverflow.com/questions/792 ... nit-matrix