Я работаю над сценарием визуализации для класса линейной алгебры в университете, и я пытаюсь показать несколько векторов, используя функцию колчана в python. Я пытаюсь построить векторы из матрицы 2x2 в одной функции колчана, однако теперь, когда я пытаюсь пометить их, я хотел бы получить доступ к каждому вектору в отдельности.
import numpy as np
import matplotlib.pyplot as plt
A = np.array([[1,3], [2,2]])
# create figure
fig = plt.figure()
# creates variable containing current figure
ax = fig.gca()
baseArrow = ax.quiver(*origin, A[0,:], A[1,:], color=['r','g']', angles='xy', scale_units='xy', scale=1)
ax.quiverkey(baseArrow,.85,.85,0.8,'i-hat',labelcolor='k',labelpos='S', coordinates = 'figure')
# display grid
plt.grid()
# display figure
plt.show()
Это позволяет мне пометить первый вектор соответствующим цветом (красным). Теперь, что я хотел бы сделать, это пометить второй вектор зеленым цветом с другой меткой? Может быть, что-то вроде:
ax.quiverkey(baseArrow**[2]**,.85,.85,0.8,'i-hat',labelcolor='k',labelpos='S', coordinates = 'figure')
Есть ли способ вытащить каждый вектор отдельно или лучше построить их отдельно, а не как вектор? Я посмотрел на следующий вопрос, но он не решает мою проблему. Диаграмма Matplotlib Quiver, соответствующая цвету метки ключа с цветом стрелки