コード例 #1
0
ファイル: swatervisual.py プロジェクト: stunax/cac
def droplet(height, width, data_type=numpy.double):
    """Generate grid of droplets"""

    x = numpy.linspace(-1, 1, num=width,
                       endpoint=True).astype(data_type)
    y = numpy.linspace(-1, 1, num=width,
                       endpoint=True).astype(data_type)

    (xx, yy) = ndgrid(x, y)

    droplet = height * numpy.exp(-5 * (xx ** 2 + yy ** 2))

    return droplet
コード例 #2
0
ファイル: swatervisual.py プロジェクト: stunax/cac
    def __init__(
        self,
        n,
        ):

        plt.ion()

        # First parameter is a figure number, second is size in inches

        self.fig = plt.figure(11, (8, 8))
        self.ax = self.fig.add_subplot(111, projection='3d')
        self.ax.set_zlim3d(-2, 3)
        self.wframe = None
        (self.X, self.Y) = ndgrid(numpy.linspace(0, n, n),
                                  numpy.linspace(0, n, n))