Numpy - заполнить изображение с corrsp. Я дал массив [U, V, I]

Question

У меня есть массив numpy с матрицей N * 3 [u, v, I]. Позиции пикселей и I интенсивность для этого пикселя.

Мне нужно сгенерировать заливку изображения соответствующего I из набора пикселей в этом массиве numpy. Прямо сейчас у меня есть для l oop, чтобы сделать это, но это довольно медленно. Какой более быстрый способ сделать это?

dmap_raw = np.zeros((raw_img_size[1], raw_img_size[0])).astype(np.float32)
for i in range(0, velodata_cam_proj.shape[0]):
    u = velodata_cam_proj[i,0]
    v = velodata_cam_proj[i,1]
    Z = velodata_cam_proj[i,2]
    dmap_raw[int(v),int(u)] = Z*100

Answer 1

Попробуйте это:

dmap_raw = np.zeros((raw_img_size[1], raw_img_size[0])).astype(np.float32)

u = velodata_cam_proj[:,0].astype('int')
v = velodata_cam_proj[:,1].astype('int')
Z = velodata_cam_proj[:,2]
dmap_raw[v, u] = Z*100