import numpy as np
import matplotlib.pyplot as plt

origin = np.array([0, 0])
v = np.array([4, 2])
w = np.array([-1, 2])
v_plus_w = v + w
v_minus_w = v - w
sum_diagonals = v_plus_w + v_minus_w

print(v)
print(w)
print(v_plus_w)
print(v_minus_w)
print(sum_diagonals)

[4 2]
[-1  2]
[3 4]
[5 0]
[8 4]

plt.quiver(
    *origin,
    *v, 
    color='r', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='v = (4, 2)'
)

plt.quiver(
    *origin,
    *w, 
    color='b', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='w = (-1, 2)'
)
plt.quiver(
    *origin,
    *v_plus_w,
    color='g',
    angles='xy',
    scale_units='xy',
    scale=1,
    label='diagonal v + w = (3, 4)'
)

plt.quiver(
    *v_minus_w,
    *v_plus_w, 
    color='g', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='diagonal v + w = (3, 4)'
)

plt.quiver(
    *origin,
    *v_minus_w, 
    color='y', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='diagonal v - w = (5, 0)'
)

plt.quiver(
    *origin,
    *sum_diagonals, 
    color='m', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='sum of diagonals = (8, 4)'
)

plt.quiver(
    *w,
    *v_minus_w, 
    color='c', 
    angles='xy', 
    scale_units='xy', 
    scale=1, 
    label='diagonal v - w = (5, 0)'
)
# Add dotted lines to form the parallelogram
plt.plot([v[0], v_plus_w[0]], [v[1], v_plus_w[1]], linestyle='dotted', color='k')
plt.plot([w[0], v_plus_w[0]], [w[1], v_plus_w[1]], linestyle='dotted', color='k')

plt.grid()
plt.xlim(-2, 6)
plt.ylim(-2, 6)
plt.axhline(y=0, color='k', linestyle='-', alpha=0.3)
plt.axvline(x=0, color='k', linestyle='-', alpha=0.3)
plt.title('2D Vectors in a Plane')
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
plt.show()