利用Python实现绘制3D爱心的代码分享

作者:神仙盼盼 时间:2021-03-30 23:25:12 

环境介绍

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()

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

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

亿点点细节

加入时间序列

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

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()

加入心脏的跳动

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

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是我们想循环的次数
       """
       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.time
           coef = np.cos(times) * (self.scale-1) + 1
           # coef 是用来放缩心脏的大小的,加入cos来使它有节奏的跳动
           self.draw(ax, coef)
           plt.pause(0.01)
           plt.show()

很好,这样我们就有了一个可以跳动的心脏,那么到这结束了嘛?

一个好的展示

当然没有!我们希望对象看到的时候他稍微有点东西,所以让它跳动却不能改变方向,岂不是看的不够全面?所以我们在加最后亿点点细节:

def run(self, count):
       fig = plt.figure()
       for i in range(count):
           plt.clf()
           ax = fig.add_subplot(projection='3d')
           ax.set_title("你对象的名字?")              # 加上你对象的小name
           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.time
           ax.view_init(10, 100+np.cos(times) * 10)   # 让三维坐标图可以变换坐标展示
           coef = np.cos(times) * (self.scale-1) + 1
           self.draw(ax, coef)
           plt.pause(0.01)  # 让绘制出来的心脏可以显示
           plt.show()

完整代码

代码完整版及效果如下

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

class Guess:
   def __init__(self, bbox=(-1.5, 1.5), resolution=50, lines=20, scale=1.2) -> None:
       self.xmin, self.xmax, self.ymin, self.ymax, self.zmin, self.zmax = bbox*3
       plt.ion()
       self.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 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, 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):
       fig = plt.figure()
       for i in range(count):
           plt.clf()
           ax = fig.add_subplot(projection='3d')
           ax.set_title("2LiuYu")
           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.time
           ax.view_init(10, 100+np.cos(times) * 10)
           coef = np.cos(times) * (self.scale-1) + 1
           self.draw(ax, coef)
           plt.pause(0.01)
           plt.show()

if __name__ == '__main__':
   demo = Guess()
   demo.run(1000)

利用Python实现绘制3D爱心的代码分享

来源:https://blog.csdn.net/qq_44961028/article/details/127778513

标签:Python,3D,爱心
0
投稿

猜你喜欢

  • python使用Thread的setDaemon启动后台线程教程

    2023-09-29 05:49:26
  • Django 大文件下载实现过程解析

    2021-12-18 20:48:32
  • python绘制已知点的坐标的直线实例

    2023-03-12 08:36:11
  • python基础之函数的返回值

    2022-10-23 23:17:06
  • html中的sub与sup标签

    2009-03-06 13:12:00
  • Python scrapy爬取起点中文网小说榜单

    2022-11-06 14:34:11
  • Python制作刷网页流量工具

    2021-10-21 00:23:28
  • 一文带你学会Mysql表批量添加字段

    2024-01-22 20:03:04
  • 一则python3的简单爬虫代码

    2022-11-29 04:46:31
  • dl,dt,dd标签 VS 传统table实现数据列表

    2009-08-02 20:45:00
  • python中return如何写

    2023-11-17 21:44:56
  • JavaScript中 this 指向问题深度解析

    2024-05-03 15:04:14
  • asp如何最大限度地实现安全登录功能?

    2010-07-11 21:11:00
  • 深度学习详解之初试机器学习

    2022-07-04 18:00:13
  • 超级链接中MailTo的语法

    2008-08-29 13:00:00
  • Python连接Redis的基本配置方法

    2023-09-29 21:28:27
  • Vue如何获取url路由地址和参数简单示例

    2024-05-09 09:53:08
  • Python使用pdb调试代码的技巧

    2022-11-18 04:36:06
  • PHP实现判断二叉树是否对称的方法

    2023-06-28 13:24:10
  • JavaScript Table行定位效果

    2009-05-25 10:47:00
  • asp之家 网络编程 m.aspxhome.com