对于颜色逐渐变化的多个图，显示 matplotlib 颜色条而不是图例

Question

我正在尝试制作一个简单的绘图，显示大量属于具有特定顺序的大型数据集的曲线，比方说绘图 1..n。曲线的形状随着 n 的增加而逐渐变化。读者能否准确看出哪个图属于哪个n值并不重要，但他们应该能够猜出n在哪个数量级。

因此我做了这样的事情：

nValues = range(0,30)
xValues = np.linspace(0,10)
dataset = [(xValues-5-0.5*n)**2 for n in nValues]
colors = {n: colorsys.hsv_to_rgb(hue,0.9,0.7) for n,hue in zip(nValues,np.linspace(0,0.7,len(nValues)))}
for n in nValues:
    plt.plot(dataset[n],color=colors[n])

（明确一点，这只是为了示例，我的数据实际上存储在一个漂亮的 pandas 数据框中。）

我想在绘图旁边添加一个颜色条，而不是图例，可能带有几个刻度线和标签（至少是最小值和最大值），以指示哪种颜色属于哪个 n 值，但我不知道如何做到这一点。我认为如果我真的从 ColorMap 中获取绘图颜色，事情可能会更容易，但我也不知道该怎么做，我也不知道如何从那里继续。

欢迎指点！

Answer 1

这几乎是其他几个问题的重复。关键是 matplotlib 需要一个 ScalarMappable 实例（通常是图像、散点图等）来制作颜色图。如果您不使用创建一个的绘图方法，则很容易伪造一个。您需要一个 Normalize 实例来定义颜色图的 min/max/scaling/etc 和一个 Colormap 实例来定义颜色。

但是，您还有一条皱纹。您没有使用颜色图，因此需要构建一个。

这是一个带有离散颜色图的示例：

import numpy as np
import matplotlib.colors as mcolors
import matplotlib.cm
import matplotlib.pyplot as plt

# Your example...
nvalues = range(0,30)
xvalues = np.linspace(0,10)

hsv2rgb = lambda hue: mcolors.hsv_to_rgb([hue,0.9,0.7])
hues = np.linspace(0, 0.7, len(nvalues))
colors = [hsv2rgb(hue) for hue in hues]

dataset = [(xvalues-5-0.5*n)**2 for n in nvalues]

fig, ax = plt.subplots()
for n in nvalues:
    ax.plot(dataset[n], color=colors[n])

# Fake a ScalarMappable so you can display a colormap
cmap, norm = mcolors.from_levels_and_colors(range(len(nvalues) + 1), colors)
sm = matplotlib.cm.ScalarMappable(cmap=cmap, norm=norm)
sm.set_array([])

fig.colorbar(sm)

plt.show()

如果您更喜欢连续的颜色图：

import numpy as np
import matplotlib.colors as mcolors
import matplotlib.cm
import matplotlib.pyplot as plt

# Your example...
nvalues = range(0,30)
xvalues = np.linspace(0,10)

hsv2rgb = lambda hue: mcolors.hsv_to_rgb([hue,0.9,0.7])
hues = np.linspace(0, 0.7, len(nvalues))
colors = [hsv2rgb(hue) for hue in hues]

dataset = [(xvalues-5-0.5*n)**2 for n in nvalues]

fig, ax = plt.subplots()
for n in nvalues:
    ax.plot(dataset[n], color=colors[n])

# Fake a ScalarMappable so you can display a colormap
cmap = mcolors.LinearSegmentedColormap.from_list('my_cmap', colors)
norm = mcolors.Normalize(min(nvalues), max(nvalues))
sm = matplotlib.cm.ScalarMappable(cmap=cmap, norm=norm)
sm.set_array([])

fig.colorbar(sm)

plt.show()

Answer 2

@tom 和@Joe Kington 都是对的：之前有人问过这个问题。但是，我试着用稍微少一点的努力来举一个例子作为链接的答案。要使用颜色图（始终将 [0,1] 中的值映射到颜色），您首先需要规范化数据。为此，您可以使用 Normalize class。 ScalarMappable 实例仅在您打算创建颜色条时才需要。

import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import matplotlib.cm as cm
import numpy as np

# your dataset
nValues = np.arange(0,30)
xValues = np.linspace(0,10)
dataset = [(xValues-5-0.5*n)**2 for n in nValues]

# setup the normalization and the colormap
normalize = mcolors.Normalize(vmin=nValues.min(), vmax=nValues.max())
colormap = cm.jet

# plot
for n in nValues:
    plt.plot(dataset[n], color=colormap(normalize(n)))

# setup the colorbar
scalarmappaple = cm.ScalarMappable(norm=normalize, cmap=colormap)
scalarmappaple.set_array(nValues)
plt.colorbar(scalarmappaple)

# show the figure
plt.show()

结果：

Answer 3

您必须创建一个新的 cmap 此代码应该有效：

import matplotlib
import matplotlib.pyplot as plt 
import colorsys
import numpy

nValues = range(30)
xValues = numpy.linspace(0,10)
dataset = [(xValues-5-0.5*n)**2 for n in nValues]
colors  = [colorsys.hsv_to_rgb(hue,0.9,0.7) for hue in numpy.linspace(0.,0.7,30)]

fig = plt.figure()
axe = fig.add_subplot(111)
trh = [axe.plot(dataset[n],color=colors[n]) for n in nValues] # one line loop for plotting the data with associated color
cma = matplotlib.colors.ListedColormap(colors, name='from_list', N=None) # creation of a user cmap
msm = matplotlib.cm.ScalarMappable(cmap=cma) 
msm.set_array(nValues) 
fig.colorbar(msm)
fig.show()

对于颜色逐渐变化的多个图，显示 matplotlib 颜色条而不是图例

show matplotlib colorbar instead of legend for multiple plots with gradually changing colors

python

matplotlib

data-visualization