python中颜色可以用在循环中吗_使用matplotlib的颜色循环作为颜色贴图

我不认为有一个用于获取当前颜色周期的公共API，但是通过模仿^{}，您可以这样定义get_prop_cycle：rcParams = plt.matplotlib.rcParams

def get_prop_cycle():

prop_cycler = rcParams['axes.prop_cycle']

if prop_cycler is None and 'axes.color_cycle' in rcParams:

clist = rcParams['axes.color_cycle']

prop_cycler = cycler('color', clist)

return prop_cycler

一旦在prop_cycler中有了颜色，就可以将c映射到颜色循环中的颜色：

^{pr2}$

import numpy as np

import matplotlib.pyplot as plt

from matplotlib.rcsetup import cycler

np.random.seed(2016)

rcParams = plt.matplotlib.rcParams

def get_prop_cycle():

prop_cycler = rcParams['axes.prop_cycle']

if prop_cycler is None and 'axes.color_cycle' in rcParams:

clist = rcParams['axes.color_cycle']

prop_cycler = cycler('color', clist)

return prop_cycler

fig, ax = plt.subplots(nrows=2)

x,y = np.random.normal(size=(2,250))

theta = np.arctan2(y,x)

c = np.digitize(theta, np.histogram(theta)[1])

ax[0].scatter(x,y,c=c)

ax[0].set_title('using default colormap')

colors = [item['color'] for item in get_prop_cycle()]

c = np.take(colors, c, mode='wrap')

ax[1].scatter(x,y,c=c)

ax[1].set_title('using color cycle')

plt.show()