13

情节线如何.plot在后续情节中重复使用？

我想在 4 个轴上绘制图，每个轴上的前三个单独的图，最后一个轴上的所有 3 个图。这是代码：

from numpy import *
from matplotlib.pyplot import *
fig=figure()
data=arange(0,10,0.01)
ax1=fig.add_subplot(2,2,1)
ax2=fig.add_subplot(2,2,2)
ax3=fig.add_subplot(2,2,3)
ax4=fig.add_subplot(2,2,4)

line1=ax1.plot(data,data)
line2=ax2.plot(data, data**2/10, ls='--', color='green')
line3=ax3.plot(data, np.sin(data), color='red')
#could I somehow use previous plots, instead recreating them all?
line4=ax4.plot(data,data)
line4=ax4.plot(data, data**2/10, ls='--', color='green')
line4=ax4.plot(data, np.sin(data), color='red')
show()

结果图片是：
在此处输入图像描述
有没有办法先定义绘图，然后将它们添加到轴上，然后绘制它们？这是我想到的逻辑：

#this is just an example, implementation can be different
line1=plot(data, data)
line2=plot(data, data**2/10, ls='--', color='green')
line3=plot(data, np.sin(data), color='red')
line4=[line1, line2, line3]

现在在 ax1 上绘制 line1，在 ax2 上绘制 line2，在 ax3 上绘制 line3，在 ax4 上绘制 line4。

4

4 回答 4

6
  • OP 中请求的实现不起作用，因为无法重用Line2D返回的绘图 Artist 。plt.plot尝试这样做，将导致RuntimeErroras perdef set_figure(self, fig):
    • line1在 OP 中，与line1直接使用Line2D方法创建的不同，因为绘制的 Artist 具有不同的属性。
    • 关于seaborn和 API matplotlib，轴级图如seaborn.lineplot返回axes：
      • p = sns.lineplot(...)然后p.get_children()获取 Artist 对象。
  • 可以使用类似的方法直接创建绘图艺术家matplotlib.lines.Line2D，并在多个绘图中重复使用。
  • 使用标准导入实践、子图更新代码，而不是使用列表理解来获得副作用（python 反模式）。
  • 经测试python 3.8.11，matplotlib 3.4.3
import numpy as np
from copy import copy
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

# crate the figure and subplots
fig, axes = plt.subplots(2, 2)

# flatten axes into 1-D for easy indexing and iteration
axes = axes.ravel()

# test data
data=np.arange(0, 10, 0.01)

# create test lines
line1 = Line2D(data, data)
line2 = Line2D(data, data**2/10, ls='--', color='green')
line3 = Line2D(data, np.sin(data), color='red')
lines = [line1, line2, line3]

# add the copies of the lines to the first 3 subplots
for ax, line in zip(axes[0:-1], lines):
    ax.add_line(copy(line))

# add 3 lines to the 4th subplot
for line in lines:
    axes[3].add_line(line)
    
# autoscale all the subplots if needed
for _a in axes:
    _a.autoscale()

plt.show()

在此处输入图像描述

原始答案

  • 这是一种可能的解决方案。我不确定它是否非常漂亮，但至少它不需要重复代码。
import numpy as np, copy
import matplotlib.pyplot as plt, matplotlib.lines as ml

fig=plt.figure(1)
data=np.arange(0,10,0.01)
ax1=fig.add_subplot(2,2,1)
ax2=fig.add_subplot(2,2,2)
ax3=fig.add_subplot(2,2,3)
ax4=fig.add_subplot(2,2,4)

#create the lines
line1=ml.Line2D(data,data)
line2=ml.Line2D(data,data**2/10,ls='--',color='green')
line3=ml.Line2D(data,np.sin(data),color='red')
#add the copies of the lines to the first 3 panels
ax1.add_line(copy.copy(line1))
ax2.add_line(copy.copy(line2))
ax3.add_line(copy.copy(line3))

[ax4.add_line(_l) for _l in [line1,line2,line3]] # add 3 lines to the 4th panel

[_a.autoscale() for _a in [ax1,ax2,ax3,ax4]] # autoscale if needed
plt.draw()
于 2012-05-08T18:01:12.433 回答
1

我认为您的用法很好，但是您可以将所有x,y数据对传递给plot这样的（尽管它使阅读变得非常可怕！）：

ax4.plot(data, data, data, data**2 / 10, data, np.sin(data))

一个有趣的不同方法是这样的：

graph_data = [(data, data), (data, data**2 / 10), (data, np.sin(data))]
[ax4.plot(i,j) for i,j in graph_data]
于 2012-05-08T14:00:48.950 回答
1

我在 jupyter notebook 中有一个更简单的用例。鉴于您在某处存储了一个图形对象，您如何重新绘制它。例如：

单元格 1：

f = plt.figure(figsize=(18, 6))
f.suptitle("Hierarchical Clustring", fontsize=20)
dendrogram(Z, color_threshold=cut_off,
           truncate_mode='lastp',
           p=20)

单元格 2：

#plot f again, the answer is really simple
f
plt.show()

而已。这样做的好处是您可以将图形存储在对象中，然后在必要时使用它们。

于 2017-12-13T09:48:27.043 回答
0

这个问题也有一个很好的例子，可以使用以下方法引用以前的轴：

fix, ax = plt.subplots(2, 2)
ax[0,1].plot(data, data**2 / 10, ls='--', color='g')

但还解释了如何使用以下方法在每个子图上插入标题：

ax[0,1].set_title('Simple plot')

ax 的维度取决于子图参数：如果它们只是水平或垂直平铺，则 ax 只需要一个索引。

于 2019-05-12T09:06:22.947 回答