且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

Python Matplotlib/底图颜色循环

更新时间:2023-11-21 23:46:40

您可以使用 ax.set_prop_cycle 指定要循环的颜色列表.

You could use ax.set_prop_cycle to specify a list of colors to cycle through.

颜色图是可调用的,因此生成颜色的一种简单方法是将 0 到 1 之间的浮点序列传递给颜色图,然后颜色图将返回一个 RGBA 颜色数组.

Colormaps are callable, so an easy way to generate colors is to pass an sequence of floats between 0 and 1 to a colormap, which will then return an array of RGBA colors.

例如,

In [93]: jet = plt.cm.jet

In [94]: jet([0,0.5,1])
Out[94]: 
array([[ 0.        ,  0.        ,  0.5       ,  1.        ],
       [ 0.49019608,  1.        ,  0.47754586,  1.        ],
       [ 0.5       ,  0.        ,  0.        ,  1.        ]])

import matplotlib.pylab as plt
import matplotlib.rcsetup as rcsetup
import numpy as np

jet = plt.cm.jet
fig, ax = plt.subplots()
N = 20
idx = np.linspace(0, 1, N)
ax.set_prop_cycle(rcsetup.cycler('color', jet(idx)))
x = np.linspace(0, 100, 200)
for i in range(1, N+1):
    ax.plot(x, np.sin(x)+i)
ax.set_ylim(0, N+1)
plt.show()