Question

我遇到了如何在模块之间共享类实例的问题。下面不是实际代码，它是我正在尝试做的简化表示。有问题的变量是thePlot。如果我将它设为全局，我会得到'全局名称未定义'类型错误。如果我不使用全局，那么形状模块中的'draw'方法找不到它。

调用draw（）方法时会出错，并尝试执行'thePlot'方法'plot'。

* Main module *****************
import matplotlib as plt
import plotter
import shapes

main():
    thePlot = plotter.plotter()
    cyl = shapes.cylinder(r, c, n, color)
    cyl.draw()
    plt.show()

* shapes module *******************
import main
import plotter

class shapes(self):
    def __init__(self):
        pass
    def cylinder(r, c, n, color):
    self.r = r
    self.c = c
    self.n = n
    self.color = color

def draw(self):
    self.x = calculate list of x coordinates
    self.y = calculate list of y coordinates
    self.z = calculate list of z coordinates
    global thePlot
    * This line causes the error
    thePlot.plot(self.x, self.y, self.z, self.color)

* plotter module ******************
import matplotlib as plt

class plotter(self):
    def __init__(self):
        self.fig = plt.figure()
        self.ax = fig.add_subplot(111, projection='3d')

    def plot(self, x, y, z, color):
        self.ax.plot_wireframe(x, y, z, color)

Answer 1

* Main module *****************
import matplotlib as plt
import plotter
import shapes

def main():
    global thePlot
    thePlot = plotter.plotter()
    cyl = shapes.cylinder(r, c, n, color)
    cyl.draw()
    plt.show()

而且：

global thePlot
* This line causes the error
thePlot.plot(self.x, self.y, self.z, self.color)

应该成为：

main.thePlot.plot(self.x, self.y, self.z, self.color)

以前调用main.main()的条件。

Answer 2

我不熟悉matplotlib，但看起来有类似的问题Add cylinder to plot。这个改编的代码怎么样？

import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D


fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

"""
def cyl():
    x = None
    y = None
    z = None
    return
"""
def cyl():
    x=np.linspace(-1, 1, 100)
    z=np.linspace(-2, 2, 100)
    Xc, Zc=np.meshgrid(x, z)
    Yc = np.sqrt(1-Xc**2)
    return Xc, Yc, Zc

Xc, Yc, Zc = cyl()

#draw parameters                                                                                                                                                          
rstride = 20
cstride = 10
ax.plot_surface(Xc, Yc, Zc, alpha=0.2, rstride=rstride, cstride=cstride)
ax.plot_surface(Xc, -Yc, Zc, alpha=0.2, rstride=rstride, cstride=cstride)

ax.set_xlabel("X")
ax.set_ylabel("Y")
ax.set_zlabel("Z")
plt.show()

如何在main中与另一个模块中的类共享实例

2 个答案: