我正在绘制一个箭头图,代码使用外部文件,如下所示:
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')
数据文件基本上看起来像:
有没有一种方法可以用不同的颜色来绘制不同的箭头方向?
Ps.:我有很多更多的箭头在我的数据文件在一个不是很合乎逻辑的句子,就像我使用的一个作为例子。
1条答案
按热度按时间slmsl1lt1#
这可能会起作用:
请注意,
plt.quiver
的第五个参数是颜色。如果你想控制颜色,你必须使用colormaps。下面是几个例子:
使用带有
colors
参数的色彩Map表:你也可以像我的第一个例子一样使用第五个参数(与
colors
相比,它的工作方式有点不同),并更改默认的colormap来控制颜色。您也可以创建自己的色彩Map表,参见例如。here。