заполнить массив numpy последовательностью чисел - PullRequest
1 голос
/ 29 февраля 2020

У меня есть массив numpy с формой (6, 3, 4), который я хотел бы заполнить возрастающей последовательностью чисел, чтобы получился такой массив:

array([[[0, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]],

   [[1, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]],

   [[2, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]],

   [[3, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]],

   [[4, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]],

   [[5, 0, 0, 0],
    [0, 0, 0, 0],
    [0, 0, 0, 0]]], dtype=uint8)

Я не хочу использовать al oop, если это возможно.

Я безуспешно пробовал следующее:

import numpy as np

new_row = np.zeros([6, 3, 4], dtype=np.uint8)
new_row[:,0:1,0:1] = range(6)

, что приводит к ValueError: could not broadcast input array from shape (6) into shape (6,1,1)

Любая помощь приветствуется

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...