Estou plotando um gráfico de seta e meu código usa um arquivo externo da seguinte maneira:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
from pylab import rcParams

data=np.loadtxt(r'data.dat')

x = data[:,0] 
y = data[:,1] 
u = data[:,2] 
v = data[:,3] 


plt.quiver(x, y, u, v, angles='xy', scale_units='xy', scale=1, pivot='mid',color='g')

O arquivo de dados basicamente se parece com:

0 0 0 1
0 1 1 0
1 0 1 0
1 1 0 1

que gera uma trama que se parece

Existe uma maneira de plotar isso com cores diferentes para as diferentes direções da seta?

Ps .: Eu tenho muito mais flechas no meu arquivo de dados em uma frase não muito lógica como a que estou usando como exemplo.