Пробовали ли вы функцию стрелки в matplotlib
( документация )?
Предполагая, что тета - это угол в радианах от оси x, возможно, что-то вроде следующего для каждой точки будетсделайте это.
arrow(x, y, cos(theta), sin(theta))
Приведенный выше код нарисует стрелку от (x, y) до (x + dx, y + dy).
Другой вариант - matplotlib.pyplot.quiver
документация . Функция quiver
позволяет нам контролировать длину стрелки разными способами.
Надеюсь, что это поможет, а не путайте!
С уважением, Prasanth