python折线图加图例_python – 将图例添加到LineCollection图

@ ubuntu的回答是正确的方法,如果你有少量的行. (如果你想添加一个传奇,你可能会这样做！)

但是,为了显示其他选项,您仍然可以使用LineCollection,您只需要使用“代理艺术家”作为图例：

import numpy as np

import matplotlib.pyplot as plt

from matplotlib.collections import LineCollection

from matplotlib.lines import Line2D

# The line format you curently have:

lines = [[(0, 1, 2, 3, 4), (4, 5, 6, 7, 8)],

[(0, 1, 2, 3, 4), (0, 1, 2, 3, 4)],

[(0, 1, 2, 3, 4), (8, 7, 6, 5, 4)],

[(4, 5, 6, 7, 8), (0, 1, 2, 3, 4)]]

# Reformat it to what `LineCollection` expects:

lines = [zip(x, y) for x, y in lines]

z = np.array([0.1, 9.4, 3.8, 2.0])

fig, ax = plt.subplots()

lines = LineCollection(lines, array=z, cmap=plt.cm.rainbow, linewidths=5)

ax.add_collection(lines)

fig.colorbar(lines)

# Manually adding artists doesn't rescale the plot, so we need to autoscale

ax.autoscale()

def make_proxy(zvalue, scalar_mappable, **kwargs):

color = scalar_mappable.cmap(scalar_mappable.norm(zvalue))

return Line2D([0, 1], [0, 1], color=color, **kwargs)

proxies = [make_proxy(item, lines, linewidth=5) for item in z]

ax.legend(proxies, ['Line 1', 'Line 2', 'Line 3', 'Line 4'])

plt.show()