vec_data_ang = np.abs(np.arctan2(dy, dx))
vec_data_avg = ndimage.convolve(np.ones(3, 3), image)
return np.dstack([vec_data_mag, vec_data_angl, vec_data_avg])
Векторизация нетривиального цикла for в numpy