Python绘制到不同图形失败

Question

编辑：我发现问题总是出现在试图绘制到两个不同的图形列表时。这是否意味着在同一个循环中不能对不同的图形列表进行绘制？请参见最新代码，了解更简单的问题示例。

我尝试分析一组复杂的数据，基本上包括在不同条件下测量电子设备的测量值。因此，代码有点复杂，但我试图将其剥离为一个可工作的示例 - 但它仍然相当长。因此，让我解释您看到的内容：您看到3个类，其中Transistor代表电子设备。它的属性Y表示测量数据 - 包括2组测量数据。每个晶体管属于一个组 - 在此示例中有2个。而一些组属于同一系列 - 在此示例中包括这两个组的一个系列。

目标现在是为每个晶体管绘制所有测量数据（未显示），然后为同一组的所有数据绘制一个图表，并将同一系列的所有数据绘制到一个图表中。为了以有效的方式编程，而不必有很多循环，我的想法是利用matplotlib的面向对象性质 - 我将为每个绘图级别（在initGrpPlt和initSeriesPlt中初始化）使用图形和子图，然后使用所有晶体管的一个循环（在MainPlt中：toGPlt和toSPlt）填充它们。最后，它应该只被打印/保存到文件/等等（PltGrp和PltSeries）。

问题：尽管我指定了要绘制的位置，但Python仍将系列图绘制到组图中。您可以通过运行带有“toSPlt（trans，j）”和不带有的代码行来自行检查此内容。我不知道为什么Python会这样做，因为在函数toSPlt中，我明确说Python应该使用系列子图列表中的子图。是否有人能想到为什么会这样，并以优雅的方式解决这个问题？

从底部向上阅读代码，这应该有助于理解。

顺祝商祺

# -*- coding: utf-8 -*-


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

maxNrVdrain = 2


X = np.linspace(-np.pi, np.pi, 256,endpoint=True)
A = [[1*np.cos(X),2*np.cos(X),3*np.cos(X),4*np.cos(X)],[1*np.tan(X),2*np.tan(X),3*np.tan(X),4*np.tan(X)]]
B = [[2* np.sin(X),4* np.sin(X),6* np.sin(X),8* np.sin(X)],[2*np.cos(X),4*np.cos(X),6*np.cos(X),8*np.cos(X)]]

class Transistor(object):
    _TransRegistry = []

    def __init__(self,y1,y2):
        self._TransRegistry.append(self)
        self.X = X
        self.Y = [y1,y2]
        self.group = ''

class Groups():
    _GroupRegistry = []

    def __init__(self,trans):
        self._GroupRegistry.append(self)
        self.transistors = [trans]
        self.figlist = []
        self.axlist = []

class Series():
    _SeriesRegistry = []

    def __init__(self,group):
        self._SeriesRegistry.append(self)
        self.groups = [group]
        self.figlist = []
        self.axlist = []


def initGrpPlt():
    for group in Groups._GroupRegistry:
        for j in range(maxNrVdrain):
            group.figlist.append(plt.figure(j))
            group.axlist.append(group.figlist[j].add_subplot(111))
    return

def initSeriesPlt():
    for series in Series._SeriesRegistry:
        for j in range(maxNrVdrain):
            series.figlist.append(plt.figure(j))
            series.axlist.append(series.figlist[j].add_subplot(111))
    return

def toGPlt(trans,j):
    colour = cm.rainbow(np.linspace(0, 1, 4))
    group = trans.group
    group.axlist[j].plot(trans.X,trans.Y[j], color=colour[group.transistors.index(trans)], linewidth=1.5, linestyle="-")
    return

def toSPlt(trans,j):
    colour = cm.rainbow(np.linspace(0, 1, 2))
    series = Series._SeriesRegistry[0]
    group = trans.group
    if group.transistors.index(trans) == 0:
        series.axlist[j].plot(trans.X,trans.Y[j],color=colour[series.groups.index(group)], linewidth=1.5, linestyle="-", label = 'T = nan, RH = nan' )
    else:
        series.axlist[j].plot(trans.X,trans.Y[j],color=colour[series.groups.index(group)], linewidth=1.5, linestyle="-")
    return

def PltGrp(group,j):
    ax = group.axlist[j]
    ax.set_title('Test Grp')
    return

def PltSeries(series,j):
    ax = series.axlist[j]
    ax.legend(loc='upper right', frameon=False)
    ax.set_title('Test Series')
    return


def MainPlt():
    initGrpPlt()
    initSeriesPlt()
    for trans in Transistor._TransRegistry:
        for j in range(maxNrVdrain):
            toGPlt(trans,j)
            toSPlt(trans,j)#plots to group plot for some reason
    for j in range(maxNrVdrain):
        for group in Groups._GroupRegistry:
            PltGrp(group,j)
    plt.show()
    return

def Init():
    for j in range(4):
        trans = Transistor(A[0][j],A[1][j])
        if j == 0:
            Groups(trans)
        else:
            Groups._GroupRegistry[0].transistors.append(trans)
        trans.group = Groups._GroupRegistry[0]
    Series(Groups._GroupRegistry[0])
    for j in range(4):
        trans = Transistor(B[0][j],B[1][j])
        if j == 0:
            Groups(trans)
        else:
            Groups._GroupRegistry[1].transistors.append(trans)
        trans.group = Groups._GroupRegistry[1]
    Series._SeriesRegistry[0].groups.append(Groups._GroupRegistry[1])
    return

def main():
    Init()
    MainPlt()
    return


main()

最近一个不可行的例子：

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



X = np.linspace(-np.pi, np.pi, 256,endpoint=True)

Y1 = np.cos(X)
Y2 = np.sin(X)

figlist1 = []
figlist2 = []

axlist1 = []
axlist2 = []



for j in range(4):
    figlist1.append(plt.figure(j))
    axlist1.append(figlist1[j].add_subplot(111))
    figlist2.append(plt.figure(j))#this should be a new set of figures!
    axlist2.append(figlist2[j].add_subplot(111))
    colour = cm.rainbow(np.linspace(0, 1, 4))
    axlist1[j].plot(X,j*Y1, color=colour[j], linewidth=1.5, linestyle="-")
    axlist1[j].set_title('Test Grp 1')
    colour = cm.rainbow(np.linspace(0, 1, 4))
    axlist2[j].plot(X,j*Y2, color=colour[int(j/2)], linewidth=1.5, linestyle="-")
    axlist2[j].set_title('Test Grp 2')
plt.show()

- Blue

Answer 1

好的，如果一个人考虑到背景就会犯愚蠢的错误，但也许有人有类似的问题，像我一样无法看到原因。所以这里是解决方案：

问题在于列表对象的名称（如figlist1 [j]）并没有定义图形 - 它们只是指向实际图形对象的指针。如果通过plt.figure（j）创建这样的对象，则必须确保j对于每个图形都不同 - 因此，在循环中初始化多个图形时，需要以某种方式更改图形的编号，否则第一个对象将被覆盖。希望能帮到你！干杯。

- Blue