cmap = plt.cm.jet
cNorm = colors.Normalize(vmin=np.min(DATA[:, (4)]), vmax=np.max(DATA[:, (4)]))
scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=cmap)
fig = plt.figure()
ax = fig.add_axes([0.1, 0.1, 0.7, 0.85])
axc = fig.add_axes([0.85, 0.1, 0.05, 0.85])
for idx in range(0, len(DATA[:, (1)])):
    colorVal = scalarMap.to_rgba(DATA[idx, 4])
ax.arrow(DATA[idx, 0], DATA[idx, 1], DATA[idx, 2] - DATA[idx, 0], DATA[idx,
    3] - DATA[idx, 1], color=colorVal)
cb1 = mpl.colorbar.ColorbarBase(axc, cmap=cmap, norm=cNorm, orientation=
    'vertical')
plt.show()
Построение стрелок разного цвета в matplotlib