programa de python que grafique un plano cartesiano y haga la funcion de translacion con matrices

import matplotlib.pyplot as plt
import numpy as np

#Vectores iniciales

A = (1,1)
B = (1,3)
C = (3,3)
D = (3,1)

A1 = np.array([1,1])
B1 = np.array([1,3])
C1 = np.array([3,3])
D1 = np.array([3,1])

#Graficar vectores

ax = plt.axes()

ax.arrow(0, 0, *A1, head_width=0.1, color ='r', head_length=0.2)
ax.arrow(0, 0, *B1, head_width=0.1, color ='b', head_length=0.2)
ax.arrow(0, 0, *C1, head_width=0.1, color ='g', head_length=0.2)
ax.arrow(0, 0, *D1, head_width=0.