如何解决Matplotlib 颤动矢量幅度用颜色表示
我试图通过用颜色替换箭头大小来减少矢量场上箭头的混乱。缩放向量不是问题(简单的除法使幅度 = 1)。我的问题是我不知道如何使用它以前的大小来改变颜色。我目前的进展或已经取得的进展是:
# Example code
# numbers used are only used as an example
import matplotlib.pyplot as plt
import numpy as np
ax = plt.subplot()
step = 7 # make arrow distribution symetrical and the number of possible arrows is the length of x_cor and y_cor
lim = 3 # to plot each arrow on a whole number (7 of them needs 7 points,-3 to 3)
x_range,y_range = np.meshgrid(np.linspace(-lim,lim,step),np.linspace(-lim,step))
x_cor = [3,5,7,1,-3,-2,0] # 1D-array
y_cor = [4,3,-2] # 1D-array
x_arrow = []
y_arrow = []
size = []
for i in range(len(x_cor)):
magnitude = np.sqrt(x_cor[i]**2 + y_cor[i]**2)
size.append(magnitude)
x_arrow.append(x_cor / magnitude)
y_arrow.append(y_cor / magnitude)
"""
Some code to convert the magnitudes to points on a possible color map?
mag_color = conversion of size[i] (magnitude for vector i) to something a colormap can interpret
"""
ax.quiver(x_range,y_range,x_arrow,y_arrow,color=mag_color)
plt.show()
我想让指示范围为暖到冷类型(红色 = 大,蓝色 = 小)并使其适用于更多点/向量/箭头。
看过其他帖子,例如link(方向部分除外),但从未真正了解如何实际执行。
预先感谢您的帮助。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。