python - 在具有单独图例的一个图中复制 Python 中的相同图

Question

我想用不同的颜色和图例在同一个图形上复制同一个图，这是我的简约工作代码。

import matplotlib.pyplot as plt
import numpy as np

def make_layout(data):
    fig = plt.figure()

    ax = fig.add_subplot(111)

    p, = ax.plot(data,'o')

    p.set_markerfacecolor('g')
    # Presumably lots of complicated settings here

    return fig, ax, p

data = data = np.linspace(0,1)

f1, a1, p1 = make_layout(data)
f2, a2, p2 = make_layout(data**2)


p2.set_markerfacecolor('yellow')

这是我在不同的图中得到的两个图，它们看起来不一样。我怎样才能做到这一点？感谢您的帮助。

我想将它们合并到同一个图中。我还想添加图例和标签。

Answer 1

您的绘图函数需要接受一个 Axes 对象作为参数。然后在函数外部创建图形和坐标轴:

import matplotlib.pyplot as plt
import numpy as np

def make_layout(data, ax):
    p, = ax.plot(data, 'o')
    p.set_markerfacecolor('g')
    return p

所以要在 1 个轴上绘制所有内容，您需要:

data = data = np.linspace(0,1)
fig, ax = plt.subplots(nrows=1, ncols=1)
p1 = make_layout(data, ax)
p2 = make_layout(data**2, ax)
p2.set_markerfacecolor('yellow')

如果你想要单独的轴，你会做这样的事情:

data = data = np.linspace(0,1)
fig, (ax1, ax2) = plt.subplots(nrows=1, ncols=2)
p1 = make_layout(data, ax1)
p2 = make_layout(data**2, ax2)
p2.set_markerfacecolor('yellow')