30岁的程序员拥有20岁的浪漫，用Python画3D玫瑰送女朋友

最新推荐文章于 2024-05-18 11:46:31 发布

原创

最新推荐文章于 2024-05-18 11:46:31 发布 · 1.2k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#python #线性代数 #经验分享

本文是一位30岁程序员分享如何利用Python的线性代数知识，画出3D玫瑰作为礼物送给女朋友的过程。通过建立柱状极坐标系，结合花瓣生长规律，运用数学公式模拟真实花瓣形状，最终用散点图逼近花瓣曲面，呈现出一朵栩栩如生的3D玫瑰。

1.天生丽质

在这里插入图片描述

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

fig = plt.figure()
ax = fig.gca(projection='3d')
# 将相位向后移动了6*pi
[x, t] = np.meshgrid(np.array(range(25)) / 24.0, np.arange(0, 575.5, 0.5) / 575 * 20 * np.pi + 4*np.pi)
p = (np.pi / 2) * np.exp(-t / (8 * np.pi))
# 添加边缘扰动
change = np.sin(15*t)/150
# 将t的参数减少，使花瓣的角度变大
u = 1 - (1 - np.mod(3.3 * t, 2 * np.pi) / np.pi) ** 4 / 2 + change
y = 2 * (x ** 2 - x) ** 2 * np.sin(p)
r = u * (x * np.sin(p) + y * np.cos(p))
h = u * (x * np.cos(p) - y * np.sin(p))
c= cm.get_cmap('Reds')
surf = ax.plot_surface(r * np.cos(t), r * np.sin(t), h, rstride=1, cstride=1,
                       cmap= c, linewidth