设为首页 加入收藏

TOP

利用Python实现绘制3D爱心的代码分享(一)
2023-07-23 13:44:49 】 浏览:39
Tags:利用 Python

环境介绍

python3.8

numpy

matplotlib

一、绘制一个三维的爱心

关于这一步,我采用的是大佬博客中的最后一种绘制方法。当然,根据我的代码习惯,我还是稍做了一点点修改的。

class Guess:
    def __init__(self, bbox=(-1.5, 1.5), resolution=50, lines=20) -> None:
        """
        bbox: 控制画格的大小
        resolution: 控制爱心的分辨率
        lines: 控制等高线的数量
        """
        self.xmin, self.xmax, self.ymin, self.ymax, self.zmin, self.zmax = bbox*3
        A = np.linspace(self.xmin, self.xmax, resolution)
        self.B = np.linspace(self.xmin, self.xmax, lines)
        self.A1, self.A2 = np.meshgrid(A, A)
        

    def coordinate(self, x, y, z):
        """
        生成坐标
        """
        return (x**2+(9/4)*y**2+z**2-1)**3-x**2*z**3-(9/80)*y**2*z**3

    def draw(self, ax):
        """
        绘制坐标
        """
        for z in self.B:
            X, Y = self.A1, self.A2
            Z = self.coordinate(X, Y, z)+z
            cset = ax.contour(X, Y, Z, [z], zdir='z', colors=('pink',))

        for y in self.B:
            X, Z = self.A1, self.A2
            Y = self.coordinate(X, y, Z)+y
            cset = ax.contour(X, Y, Z, [y], zdir='y', colors=('pink',))

        for x in self.B:
            Y, Z = self.A1, self.A2
            X = self.coordinate(x, Y, Z) + x
            cset = ax.contour(X, Y, Z, [x], zdir='x', colors=('pink',))

    def run(self):
        fig = plt.figure()
        ax = fig.add_subplot(projection='3d')
        ax.set_zlim3d(self.zmin, self.zmax)
        ax.set_xlim3d(self.xmin, self.xmax)
        ax.set_ylim3d(self.ymin, self.ymax)
        plt.show()

但是这可以达到我们想要的效果吗?

显然不能!于是我们开始加入亿点点细节!

二、细节点

1.加入时间序列

想要心脏跳起来,我们就需要有时间维度的变化。那怎么做最合理呢?这里仅展示修改的代码位置。

代码如下(示例):

class Guess:
    def __init__(self, bbox=(-1.5, 1.5), resolution=50, lines=20) -> None:
        plt.ion()                                         # 开启画布的动态图模式
        self.xmin, self.xmax, self.ymin, self.ymax, self.zmin, self.zmax = bbox*3
        self.time = time.time()                           # 这里有一个衡量的时间坐标,很合理吧
        A = np.linspace(self.xmin, self.xmax, resolution)
        self.B = np.linspace(self.xmin, self.xmax, lines)
        self.A1, self.A2 = np.meshgrid(A, A)

    def run(self, count):
        """
        加入count是我们想循环的次数
        """
        fig = plt.figure()
        for i in range(count):
            plt.clf()                               # 每次清除画布
            ax = fig.add_subplot(projection='3d')
            ax.set_zlim3d(self.zmin, self.zmax)
            ax.set_xlim3d(self.xmin, self.xmax)
            ax.set_ylim3d(self.ymin, self.ymax)
            times = time.time()-self.t/ime          # 计算画布的当前时间状态
            self.draw(ax, coef)
            plt.show()

 

2.加入心脏的跳动

心脏的跳动当然不会是线性的了,我们需要心脏的跳动是有层次感的,并且还是可以做往返运动的。

emmmmm… 这么说来,cos是不是就是做这个用的?

于是…
代码如下(示例):

def __init__(self, bbox=(-1.5, 1.5), resolution=50, lines=20, scale=1.2) ->
 None:
        """
        scale: 心脏缩放的系数
        """
        self.xmin, self.xmax, self.ymin, self.ymax, self.zmin, self.zmax = bbox*3
        plt.ion() 
        self.scale = scale   # scale: 心脏缩放的系数 设置为全局变量
        self.time = time.time()
        A = np.linspace(self.xmin, self.xmax, resolution)
        self.B = np.linspace(self.xmin, self.xmax, lines)
        self.A1, self.A2 = np.meshgrid(A, A)

    def draw(self, ax, coef):
        """
        coef: 使得心脏可以按照时间跳动
        """
        for z in self.B:
            X, Y = self.A1, self.A2
            Z = self.coordinate(X, Y, z)+z
            cset = ax.contour(X * coef, Y * coef, Z * coef, [z * coef], zdir='z', colors=('pink',))

        for y in self.B:
            X, Z = self.A1, self.A2
            Y = self.coordinate(X, y, Z)+y
            cset = ax.contour(X * coef, Y * coef, Z * coef, [y * coef], zdir='y', colors=('pink',))

        for x in self.B:
            Y, Z = self.A1, self.A2
            X = self.coordinate(x, Y, Z) + x
            cset = ax.contour(X * coef, Y * coef, Z * coef, [x * coef], zdir='x', colors=('pink',))

    def run(self, count):
        """
        加入count是我们想循环的次数
        &q
首页 上一页 1 2 下一页 尾页 1/2/2
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇Python爬虫采集商品评价信息--京东 下一篇Python初学者必备技能,使用Pytho..

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目