Python对头发二维建模(考虑风力、重力)

2024-03-12 01:44

本文主要是介绍Python对头发二维建模(考虑风力、重力),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

一、背景

二、代码


一、背景

数值方法被用于创建电影、游戏或其他媒体中的计算机图形。例如,生成“逼真”的烟雾、水或爆炸等动画。本文内容是对头发的模拟,要求考虑重力、风力的影响。

假设:
1、人的头部是一个半径为10厘米的球体。
2、每根头发都与球体的表面垂直相交。
3、作用在每根头发上的力包括重力(在-z方向上)和恒定的风力(在+x方向上)。

二、代码

#导入python包
import numpy as np
import matplotlib.pyplot as plt
import scipy.integrate
import scipy.optimizedef rhs_func_wrapper(f_x, f_g):'''输入:f_x风力、f_g重力输出:函数rhs_func,用于包装常微分方程'''def rhs_func(s, y):'''输入:s弧度(自变量)y即[角度θ,梯度u](因变量)'''theta = y[0]u = y[1]dyds = np.zeros_like(y)dyds[0] = u  #一阶导dyds[1] = s * f_g * np.cos(theta) + s * f_x * np.sin(theta) #二阶常微分方程,对应方程(3a)return dydsreturn rhs_funcdef shot(u0, theta_0, L, rhs_func):'''解决边界值问题(BVP)返回:s弧长、y包含角度和梯度的数组、sol是OdeSolution对象,表示常微分方程的解(描述弧长s和角度θ之间关系)'''y0 = np.array([theta_0, u0])interval = [0, L] solution = scipy.integrate.solve_ivp(rhs_func,  interval, #rhs_func中参数s的范围y0,  #初始条件max_step=1e-2, #设置步长dense_output=True)  #用于生成sol,可以用于在任意点插值解s, y, sol = solution.t, solution.y, solution.solreturn s, y, soldef shot_error_wrapper(theta_0, L, rhs_func):'''计算误差'''def shot_error(u0):s, y, sol = shot(u0, theta_0, L, rhs_func)phi = y[1, -1] #提取二维数组y中的梯度的最后一个元素,作为误差return phireturn shot_errordef coordinate_rhs_func_wrapper(theta_s):'''计算头发坐标的导数输入:theta_s表示一个描述弧长s和角度θ之间关系的OdeSolution对象'''def coordinate_rhs_func(s, y):'''输入:弧长s、y表示坐标(x,z)'''dyds = np.zeros_like(y) #初始化一个与y相同大小的数组dyds,用于存储导数theta = theta_s(s)[0]  #计算弧长s对应的角度theta,通过调用theta_s(s)获取,并取得返回值的第一个元素dyds[0] = np.cos(theta)  #求导公式dyds[1] = np.sin(theta)  #求导公式return dydsreturn coordinate_rhs_funcdef hair_bvp_2d(theta_0_list, L, R, f_x, f_g=0.1):'''输入:theta_0_list初始角度列表,L头发长度,R人头半径,f_x风力,f_g重力(默认为0.1)'''rhs_func = rhs_func_wrapper(f_x, f_g)x_list = [] #初始化两个空列表用于存储解z_list = []for theta_0 in theta_0_list:  #对于每根头发的初始角度theta_0进行以下步骤shot_error = shot_error_wrapper(theta_0, L, rhs_func)u0 = scipy.optimize.brentq(shot_error, -10, 10)  #在-10~10区间内找到误差最小的初始梯度u0s, y, sol = shot(u0, theta_0, L, rhs_func)coordinate_rhs_func = coordinate_rhs_func_wrapper(sol)y0 = np.array([R * np.cos(theta_0), R * np.sin(theta_0)])  #设置初始条件interval = [0, L]solution = scipy.integrate.solve_ivp(coordinate_rhs_func, interval, y0,max_step=1e-2)x_list.append(solution.y[0]) #402个横坐标z_list.append(solution.y[1]) #402个纵坐标x = np.array(x_list)z = np.array(z_list)return x, zdef plot_hairs(x, z, R, title):#画人头:半径为10的圆,颜色为bluetheta_list = np.linspace(0, 2 * np.pi, 50)x_head = R * np.cos(theta_list)y_head = R * np.sin(theta_list)plt.plot(x_head, y_head, c='blue')#依次画每根头发,颜色为grayfor i in range(x.shape[0]): x_coords = x[i, :]z_coords = z[i, :]plt.plot(x_coords, z_coords, c='gray')ax = plt.gca()  #获取坐标轴实例ax.set_aspect(1) #纵横单位长度比例为1:1plt.xlabel('x') #横坐标名称plt.ylabel('z') #纵坐标名称plt.title(title) #图的名称plt.show()  #打印出来if __name__ == "__main__":L = 4  #头发长度:4cmR = 10  #人的头部,半径10cmtheta_0_list = np.linspace(0, np.pi, 20)  #0-π按20等分切分print('Task 1 - no gravity')x, z = hair_bvp_2d(theta_0_list, L, R, 0, 0)assert x.shape[0] == 20 and z.shape[0] == 20 and x.shape[1] == z.shape[1]  #断言,如果不满足条件,则中断程序plot_hairs(x, z, R, title='Task 1 - no gravity') #生成图像print('Task 2 - no wind')x, z = hair_bvp_2d(theta_0_list, L, R, 0)assert x.shape[0] == 20 and z.shape[0] == 20 and x.shape[1] == z.shape[1] plot_hairs(x, z, R, title='Task 2 - no wind') print('Task 3 - wind (f_x=0.1)')x, z = hair_bvp_2d(theta_0_list, L, R, 0.1)assert x.shape[0] == 20 and z.shape[0] == 20 and x.shape[1] == z.shape[1]plot_hairs(x, z, R, title='Task 3 - wind (f_x=0.1)')

运行结果:

无重力、无风力
有重力、无风力
有重力、有风力

这篇关于Python对头发二维建模(考虑风力、重力)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/799763

相关文章

Python将博客内容html导出为Markdown格式

《Python将博客内容html导出为Markdown格式》Python将博客内容html导出为Markdown格式,通过博客url地址抓取文章,分析并提取出文章标题和内容,将内容构建成html,再转... 目录一、为什么要搞?二、准备如何搞?三、说搞咱就搞!抓取文章提取内容构建html转存markdown

Python获取中国节假日数据记录入JSON文件

《Python获取中国节假日数据记录入JSON文件》项目系统内置的日历应用为了提升用户体验,特别设置了在调休日期显示“休”的UI图标功能,那么问题是这些调休数据从哪里来呢?我尝试一种更为智能的方法:P... 目录节假日数据获取存入jsON文件节假日数据读取封装完整代码项目系统内置的日历应用为了提升用户体验,

Python FastAPI+Celery+RabbitMQ实现分布式图片水印处理系统

《PythonFastAPI+Celery+RabbitMQ实现分布式图片水印处理系统》这篇文章主要为大家详细介绍了PythonFastAPI如何结合Celery以及RabbitMQ实现简单的分布式... 实现思路FastAPI 服务器Celery 任务队列RabbitMQ 作为消息代理定时任务处理完整

Python Websockets库的使用指南

《PythonWebsockets库的使用指南》pythonwebsockets库是一个用于创建WebSocket服务器和客户端的Python库,它提供了一种简单的方式来实现实时通信,支持异步和同步... 目录一、WebSocket 简介二、python 的 websockets 库安装三、完整代码示例1.

揭秘Python Socket网络编程的7种硬核用法

《揭秘PythonSocket网络编程的7种硬核用法》Socket不仅能做聊天室,还能干一大堆硬核操作,这篇文章就带大家看看Python网络编程的7种超实用玩法,感兴趣的小伙伴可以跟随小编一起... 目录1.端口扫描器:探测开放端口2.简易 HTTP 服务器:10 秒搭个网页3.局域网游戏:多人联机对战4.

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

Python使用自带的base64库进行base64编码和解码

《Python使用自带的base64库进行base64编码和解码》在Python中,处理数据的编码和解码是数据传输和存储中非常普遍的需求,其中,Base64是一种常用的编码方案,本文我将详细介绍如何使... 目录引言使用python的base64库进行编码和解码编码函数解码函数Base64编码的应用场景注意

Python基于wxPython和FFmpeg开发一个视频标签工具

《Python基于wxPython和FFmpeg开发一个视频标签工具》在当今数字媒体时代,视频内容的管理和标记变得越来越重要,无论是研究人员需要对实验视频进行时间点标记,还是个人用户希望对家庭视频进行... 目录引言1. 应用概述2. 技术栈分析2.1 核心库和模块2.2 wxpython作为GUI选择的优

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.

Python+PyQt5实现多屏幕协同播放功能

《Python+PyQt5实现多屏幕协同播放功能》在现代会议展示、数字广告、展览展示等场景中,多屏幕协同播放已成为刚需,下面我们就来看看如何利用Python和PyQt5开发一套功能强大的跨屏播控系统吧... 目录一、项目概述:突破传统播放限制二、核心技术解析2.1 多屏管理机制2.2 播放引擎设计2.3 专