У меня есть массив 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)
Любая помощь приветствуется