gpt4 book ai didi

python - 如何在此 matplotlib 图中使用颜色图?

转载 作者:太空宇宙 更新时间:2023-11-04 04:23:34 25 4
gpt4 key购买 nike

我正在尝试向此 matplotlib 绘图添加颜色图(代码如下)。我使用颜色渐变来说明运动在时间或步骤中的进展(色调是时间或 n 的函数)。现在我想在图中的插图中添加一个颜色图,但我无法成功。

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
plt.style.use('ggplot')
n = 5000
x = np.cumsum(np.random.randn(n))
y = np.cumsum(np.random.randn(n))
# We add 10 intermediary points between two successive points. We
# interpolate x and y.
k = 10
x2 = np.interp(np.arange(n * k), np.arange(n) * k, x)
y2 = np.interp(np.arange(n * k), np.arange(n) * k, y)
fig, ax = plt.subplots(1, 1, figsize=(8, 8))
# Now, we draw our points with a gradient of colors.
ax.scatter(x2, y2, c=range(n * k), linewidths=0, marker='o', s=3,
cmap=plt.cm.jet,)
ax.axis('equal')
ax.set_axis_off()

我正在寻找的是绘图右侧的颜色条,其比例从 0 到 5000 均等。

最佳答案

您需要为 colorbar 提供轴句柄

ax_ = ax.scatter(x2, y2, c=range(n * k), linewidths=0, marker='o', s=3, cmap=plt.cm.jet,) # <---- store plot instance in ax_
ax.axis('equal')

plt.colorbar(ax_) # <--- here provide ax_ (can choose other name as well)
ax.set_axis_off()

enter image description here

关于python - 如何在此 matplotlib 图中使用颜色图?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53954292/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com