Различных цветов для стрелок в колчане участке

Question

Я черчение стрелка график и мой код использует внешний файл следующим образом:

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') 

Файл данных в основном выглядит следующим образом:

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

, который генерирует сюжет, который выглядит как

есть Aw ay построить это с разными цветами для разных направлений стрелок?

Пс .: У меня есть гораздо больше стрел в моем файле данных в не очень логичном предложении, как тот, который я использую в качестве примера.

Answer 1

Это, вероятно, сделать трюк:

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

Обратите внимание, что аргумент в пятом по plt.quiver является цветом.

---

UPD. Если вы хотите управлять цветами, вы должны использовать colormaps. Вот несколько примеров:

Используйте палитру с colors параметра:

import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
from matplotlib.colors import Normalize 

%matplotlib inline 

ph = np.linspace(0, 2*np.pi, 13) 
x = np.cos(ph) 
y = np.sin(ph) 
u = np.cos(ph) 
v = np.sin(ph) 
colors = arctan2(u, v) 

norm = Normalize() 
norm.autoscale(colors) 
# we need to normalize our colors array to match it colormap domain 
# which is [0, 1] 

colormap = cm.inferno 
# pick your colormap here, refer to 
# http://matplotlib.org/examples/color/colormaps_reference.html 
# and 
# http://matplotlib.org/users/colormaps.html 
# for details 
plt.figure(figsize=(6, 6)) 
plt.xlim(-2, 2) 
plt.ylim(-2, 2) 
plt.quiver(x, y, u, v, color=colormap(norm(colors)), angles='xy', 
      scale_units='xy', scale=1, pivot='mid') 

Вы также можете придерживаться пятый аргумент, как в моем первом примере (который работает в несколько иной способ сравнения с colors) и изменить цветовую карту по умолчанию для управления цветами.

plt.rcParams['image.cmap'] = 'Paired' 

plt.figure(figsize=(6, 6)) 
plt.xlim(-2, 2) 
plt.ylim(-2, 2) 

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

Вы также можете создавать свои собственные цветовые карты, смотри, например, here.