Python 旋转立方体

2024-05-13 12:44
文章标签 python 立方体 旋转

本文主要是介绍Python 旋转立方体,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

文章目录

  • 效果图
  • 运行环境
  • 完整代码
  • 实现思路
    • 1. 导入库和定义常量
    • 2. 创建Cube类
    • 3. 实现Cube类的draw方法
    • 4. 实现主函数


效果图

请添加图片描述

运行环境

python版本:python3.x

依赖包:

$ pip install pygame
$ pip install numpy

完整代码

import numpy as np  # 导入 NumPy 库,用于数值计算和处理多维数组
import pygame  # 导入 Pygame 库,用于游戏开发和图形界面设计# 定义屏幕的宽度和高度
WIDTH = 800
HEIGHT = 800# 定义颜色常量
BLACK = (0, 0, 0)  # 黑色
WHITE = (255, 255, 255)  # 白色class Cube:"""表示一个立方体。"""def __init__(self, pos: np.ndarray, a: float) -> None:"""初始化立方体。:param pos: 立方体的中心位置,是一个包含三个元素的 NumPy 数组。:param a: 立方体的边长。"""self.pos = pos  # 立方体的中心位置self.angle = np.pi / 4  # 立方体的旋转角度,初始化为 45 度self.center_offset = np.array([-a / 2, -a / 2, -a / 2])  # 立方体顶点到中心的偏移量self.edges = np.array([  # 立方体的边,是一个包含 12 条边的数组# 前脸的四条边np.array([np.array([0, 0, 0]), np.array([a, 0, 0])]),np.array([np.array([a, 0, 0]), np.array([a, a, 0])]),np.array([np.array([a, a, 0]), np.array([0, a, 0])]),np.array([np.array([0, a, 0]), np.array([0, 0, 0])]),# 右脸的四条边np.array([np.array([0, 0, 0]), np.array([0, 0, a])]),np.array([np.array([a, a, 0]), np.array([a, a, a])]),np.array([np.array([a, 0, 0]), np.array([a, 0, a])]),np.array([np.array([0, a, 0]), np.array([0, a, a])]),# 上脸的四条边np.array([np.array([0, 0, a]), np.array([a, 0, a])]),np.array([np.array([a, 0, a]), np.array([a, a, a])]),np.array([np.array([a, a, a]), np.array([0, a, a])]),np.array([np.array([0, a, a]), np.array([0, 0, a])]),])def draw(self, screen: pygame.surface.Surface, rotation_rate: float) -> None:"""在屏幕上绘制立方体。:param screen: 要绘制立方体的 Pygame 屏幕对象。:param rotation_rate: 立方体的旋转速率,用于控制立方体旋转的速度。"""# 将立方体的边加上中心偏移量,得到实际的顶点位置rotated_cube = np.add(self.edges, self.center_offset)# 计算绕 X、Y、Z 轴旋转的矩阵rotation_matrix_x = np.array([[1, 0, 0],[0, np.cos(self.angle), -np.sin(self.angle)],[0, np.sin(self.angle), np.cos(self.angle)]])rotation_matrix_y = np.array([[np.cos(self.angle), 0, np.sin(self.angle)],[0, 1, 0],[-np.sin(self.angle), 0, np.cos(self.angle)]])rotation_matrix_z = np.array([[np.cos(self.angle), -np.sin(self.angle), 0],[np.sin(self.angle), np.cos(self.angle), 0],[0, 0, 1],])# 对立方体进行旋转rotated_cube = np.matmul(rotated_cube, rotation_matrix_x)rotated_cube = np.matmul(rotated_cube, rotation_matrix_y)rotated_cube = np.matmul(rotated_cube, rotation_matrix_z)# 将旋转后的立方体移动到正确的位置moved_cube = np.add(self.pos, rotated_cube)# 在屏幕上绘制立方体的边for edge in moved_cube:# 获取边的两个端点的屏幕坐标start_pos = edge[0][0:2]end_pos = edge[1][0:2]# 绘制边pygame.draw.line(screen, WHITE, start_pos, end_pos)# 更新立方体的旋转角度self.angle += rotation_ratedef main():"""主函数,启动 Pygame 并创建旋转的立方体。"""# 初始化 Pygamepygame.init()# 创建屏幕对象screen = pygame.display.set_mode((WIDTH, HEIGHT))# 设置窗口标题pygame.display.set_caption("旋转立方体 By stormsha")# 创建立方体对象,中心位于 (400, 400, 200),边长为 200cube = Cube(np.array([400, 400, 200]), 200)# 主循环running = Truewhile running:# 处理 Pygame 事件,如关闭窗口等for event in pygame.event.get():if event.type == pygame.QUIT:running = False# 清空屏幕screen.fill(BLACK)# 绘制立方体cube.draw(screen, 0.001)# 更新屏幕pygame.display.flip()if __name__ == "__main__":# 如果脚本被直接运行,则执行主函数main()

实现思路

使用Pygame和NumPy创建旋转立方体

在这篇文章中,我们将详细介绍如何使用Python的Pygame库和NumPy库创建一个旋转的立方体。我们将逐步讲解代码的实现思路,并解释关键部分的作用。

1. 导入库和定义常量

首先,我们需要导入所需的库,包括NumPy和Pygame。NumPy库用于数值计算和处理多维数组,而Pygame库用于游戏开发和图形界面设计。

import numpy as np  # 导入 NumPy 库,用于数值计算和处理多维数组
import pygame  # 导入 Pygame 库,用于游戏开发和图形界面设计# 定义屏幕的宽度和高度
WIDTH = 800
HEIGHT = 800# 定义颜色常量
BLACK = (0, 0, 0)  # 黑色
WHITE = (255, 255, 255)  # 白色

2. 创建Cube类

接下来,我们创建一个名为Cube的类,用于表示立方体。在Cube类的构造函数中,我们初始化立方体的中心位置、旋转角度和边长。

class Cube:"""表示一个立方体。"""def __init__(self, pos: np.ndarray, a: float) -> None:"""初始化立方体。:param pos: 立方体的中心位置,是一个包含三个元素的 NumPy 数组。:param a: 立方体的边长。"""self.pos = pos  # 立方体的中心位置self.angle = np.pi / 4  # 立方体的旋转角度,初始化为 45 度self.center_offset = np.array([-a / 2, -a / 2, -a / 2])  # 立方体顶点到中心的偏移量self.edges = np.array([  # 立方体的边,是一个包含 12 条边的数组# 前脸的四条边np.array([np.array([0, 0, 0]), np.array([a, 0, 0])]),np.array([np.array([a, 0, 0]), np.array([a, a, 0])]),np.array([np.array([a, a, 0]), np.array([0, a, 0])]),np.array([np.array([0, a, 0]), np.array([0, 0, 0])]),# 右脸的四条边np.array([np.array([0, 0, 0]), np.array([0, 0, a])]),np.array([np.array([a, a, 0]), np.array([a, a, a])]),np.array([np.array([a, 0, 0]), np.array([a, 0, a])]),np.array([np.array([0, a, 0]), np.array([0, a, a])]),# 上脸的四条边np.array([np.array([0, 0, a]), np.array([a, 0, a])]),np.array([np.array([a, 0, a]), np.array([a, a, a])]),np.array([np.array([a, a, a]), np.array([0, a, a])]),np.array([np.array([0, a, a]), np.array([0, 0, a])]),])

在上面的代码中,我们定义了立方体的12条边,包括前脸、右脸和上脸的边。每条边由两个顶点组成,使用NumPy的数组表示。

3. 实现Cube类的draw方法

接下来,我们实现Cube类的draw方法,用于在屏幕上绘制立方体。在draw方法中,我们首先将立方体的边加上中心偏移量,得到实际的顶点位置。然后,我们计算绕X、Y、Z轴旋转的矩阵,并对立方体进行旋转。最后,我们将旋转后的立方体移动到正确的位置,并在屏幕上绘制立方体的边。

def draw(self, screen: pygame.surface.Surface, rotation_rate: float) -> None:"""在屏幕上绘制立方体。:param screen: 要绘制立方体的 Pygame 屏幕对象。:param rotation_rate: 立方体的旋转速率,用于控制立方体旋转的速度。"""# 将立方体的边加上中心偏移量,得到实际的顶点位置rotated_cube = np.add(self.edges, self.center_offset)# 计算绕 X、Y、Z 轴旋转的矩阵rotation_matrix_x = np.array([[1, 0, 0],[0, np.cos(self.angle), -np.sin(self.angle)],[0, np.sin(self.angle), np.cos(self.angle)]])rotation_matrix_y = np.array([[np.cos(self.angle), 0, np.sin(self.angle)],[0, 1, 0],[-np.sin(self.angle), 0, np.cos(self.angle)]])rotation_matrix_z = np.array([[np.cos(self.angle), -np.sin(self.angle), 0],[np.sin(self.angle), np.cos(self.angle), 0],[0, 0, 1],])# 对立方体进行旋转rotated_cube = np.matmul(rotated_cube, rotation_matrix_x)rotated_cube = np.matmul(rotated_cube, rotation_matrix_y)rotated_cube = np.matmul(rotated_cube, rotation_matrix_z)# 将旋转后的立方体移动到正确的位置moved_cube = np.add(self.pos, rotated_cube)# 在屏幕上绘制立方体的边for edge in moved_cube:# 获取边的两个端点的屏幕坐标start_pos = edge[0][0:2]end_pos = edge[1][0:2]# 绘制边pygame.draw.line(screen, WHITE, start_pos, end_pos)# 更新立方体的旋转角度self.angle += rotation_rate

在上面的代码中,我们使用NumPy的matmul函数进行矩阵乘法,以实现立方体的旋转。然后,我们使用pygame的draw.line函数在屏幕上绘制立方体的边。

4. 实现主函数

最后,我们实现主函数main,用于启动Pygame并创建旋转的立方体。在主函数中,我们首先初始化Pygame,并创建屏幕对象。然后,我们创建一个Cube对象,并进入主循环。在主循环中,我们处理Pygame事件,清空屏幕,绘制立方体,并更新屏幕。

def main():"""主函数,启动 Pygame 并创建旋转的立方体。"""# 初始化 Pygamepygame.init()# 创建屏幕对象screen = pygame.display.set_mode((WIDTH, HEIGHT))# 设置窗口标题pygame.display.set_caption("旋转立方体 By stormsha")# 创建立方体对象,中心位于 (400, 400, 200),边长为 200cube = Cube(np.array([400, 400, 200]), 200)# 主循环running = Truewhile running:# 处理 Pygame 事件,如关闭窗口等for event in pygame.event.get():if event.type == pygame.QUIT:running = False# 清空屏幕screen.fill(BLACK)# 绘制立方体cube.draw(screen, 0.001)# 更新屏幕pygame.display.flip()if __name__ == "__main__":# 如果脚本被直接运行,则执行主函数main()

在上面的代码中,我们使用pygame的event.get函数获取事件,并判断是否为关闭窗口事件。如果是关闭窗口事件,则退出主循环。然后,我们使用screen.fill函数清空屏幕,并使用cube.draw函数绘制立方体。最后,我们使用pygame的display.flip函数更新屏幕。

这篇关于Python 旋转立方体的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



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

相关文章

Python实现图片分割的多种方法总结

《Python实现图片分割的多种方法总结》图片分割是图像处理中的一个重要任务,它的目标是将图像划分为多个区域或者对象,本文为大家整理了一些常用的分割方法,大家可以根据需求自行选择... 目录1. 基于传统图像处理的分割方法(1) 使用固定阈值分割图片(2) 自适应阈值分割(3) 使用图像边缘检测分割(4)

一文带你搞懂Python中__init__.py到底是什么

《一文带你搞懂Python中__init__.py到底是什么》朋友们,今天我们来聊聊Python里一个低调却至关重要的文件——__init__.py,有些人可能听说过它是“包的标志”,也有人觉得它“没... 目录先搞懂 python 模块(module)Python 包(package)是啥?那么 __in

使用Python实现图像LBP特征提取的操作方法

《使用Python实现图像LBP特征提取的操作方法》LBP特征叫做局部二值模式,常用于纹理特征提取,并在纹理分类中具有较强的区分能力,本文给大家介绍了如何使用Python实现图像LBP特征提取的操作方... 目录一、LBP特征介绍二、LBP特征描述三、一些改进版本的LBP1.圆形LBP算子2.旋转不变的LB

Python中__init__方法使用的深度解析

《Python中__init__方法使用的深度解析》在Python的面向对象编程(OOP)体系中,__init__方法如同建造房屋时的奠基仪式——它定义了对象诞生时的初始状态,下面我们就来深入了解下_... 目录一、__init__的基因图谱二、初始化过程的魔法时刻继承链中的初始化顺序self参数的奥秘默认

Python实现特殊字符判断并去掉非字母和数字的特殊字符

《Python实现特殊字符判断并去掉非字母和数字的特殊字符》在Python中,可以通过多种方法来判断字符串中是否包含非字母、数字的特殊字符,并将这些特殊字符去掉,本文为大家整理了一些常用的,希望对大家... 目录1. 使用正则表达式判断字符串中是否包含特殊字符去掉字符串中的特殊字符2. 使用 str.isa

python中各种常见文件的读写操作与类型转换详细指南

《python中各种常见文件的读写操作与类型转换详细指南》这篇文章主要为大家详细介绍了python中各种常见文件(txt,xls,csv,sql,二进制文件)的读写操作与类型转换,感兴趣的小伙伴可以跟... 目录1.文件txt读写标准用法1.1写入文件1.2读取文件2. 二进制文件读取3. 大文件读取3.1

使用Python实现一个优雅的异步定时器

《使用Python实现一个优雅的异步定时器》在Python中实现定时器功能是一个常见需求,尤其是在需要周期性执行任务的场景下,本文给大家介绍了基于asyncio和threading模块,可扩展的异步定... 目录需求背景代码1. 单例事件循环的实现2. 事件循环的运行与关闭3. 定时器核心逻辑4. 启动与停

基于Python实现读取嵌套压缩包下文件的方法

《基于Python实现读取嵌套压缩包下文件的方法》工作中遇到的问题,需要用Python实现嵌套压缩包下文件读取,本文给大家介绍了详细的解决方法,并有相关的代码示例供大家参考,需要的朋友可以参考下... 目录思路完整代码代码优化思路打开外层zip压缩包并遍历文件:使用with zipfile.ZipFil

Python处理函数调用超时的四种方法

《Python处理函数调用超时的四种方法》在实际开发过程中,我们可能会遇到一些场景,需要对函数的执行时间进行限制,例如,当一个函数执行时间过长时,可能会导致程序卡顿、资源占用过高,因此,在某些情况下,... 目录前言func-timeout1. 安装 func-timeout2. 基本用法自定义进程subp

Python实现word文档内容智能提取以及合成

《Python实现word文档内容智能提取以及合成》这篇文章主要为大家详细介绍了如何使用Python实现从10个左右的docx文档中抽取内容,再调整语言风格后生成新的文档,感兴趣的小伙伴可以了解一下... 目录核心思路技术路径实现步骤阶段一:准备工作阶段二:内容提取 (python 脚本)阶段三:语言风格调