import matplotlib.pyplot as plt
import numpy as np
fig, ax = plt.subplots()
x = np.linspace(0, 2, 1000)
y = x ** 2
ax.plot(x, y)
x0, y0 = 1.0, 1.0
dx, dy = 1, 2
length = 1.25
dx, dy = length * np.array([dx, dy]) / np.hypot(dx, dy)
ax.quiver(x0, y0, dx, dy, units='inches', angles='xy', scale=1, scale_units
    ='inches', color='red')
ax.axis((0, 2, 0, 2))
plt.show()
Направление стрелки задается данными, а длина задается размером фигуры