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脚本实现自动删除C盘临时文件夹

《Python脚本实现自动删除C盘临时文件夹》在日常使用电脑的过程中,临时文件夹往往会积累大量的无用数据,占用宝贵的磁盘空间,下面我们就来看看Python如何通过脚本实现自动删除C盘临时文件夹吧... 目录一、准备工作二、python脚本编写三、脚本解析四、运行脚本五、案例演示六、注意事项七、总结在日常使用

Python将大量遥感数据的值缩放指定倍数的方法(推荐)

《Python将大量遥感数据的值缩放指定倍数的方法(推荐)》本文介绍基于Python中的gdal模块,批量读取大量多波段遥感影像文件,分别对各波段数据加以数值处理,并将所得处理后数据保存为新的遥感影像... 本文介绍基于python中的gdal模块,批量读取大量多波段遥感影像文件,分别对各波段数据加以数值处

python管理工具之conda安装部署及使用详解

《python管理工具之conda安装部署及使用详解》这篇文章详细介绍了如何安装和使用conda来管理Python环境,它涵盖了从安装部署、镜像源配置到具体的conda使用方法,包括创建、激活、安装包... 目录pytpshheraerUhon管理工具:conda部署+使用一、安装部署1、 下载2、 安装3

Python进阶之Excel基本操作介绍

《Python进阶之Excel基本操作介绍》在现实中,很多工作都需要与数据打交道,Excel作为常用的数据处理工具,一直备受人们的青睐,本文主要为大家介绍了一些Python中Excel的基本操作,希望... 目录概述写入使用 xlwt使用 XlsxWriter读取修改概述在现实中,很多工作都需要与数据打交

使用Python实现在Word中添加或删除超链接

《使用Python实现在Word中添加或删除超链接》在Word文档中,超链接是一种将文本或图像连接到其他文档、网页或同一文档中不同部分的功能,本文将为大家介绍一下Python如何实现在Word中添加或... 在Word文档中,超链接是一种将文本或图像连接到其他文档、网页或同一文档中不同部分的功能。通过添加超

Python MySQL如何通过Binlog获取变更记录恢复数据

《PythonMySQL如何通过Binlog获取变更记录恢复数据》本文介绍了如何使用Python和pymysqlreplication库通过MySQL的二进制日志(Binlog)获取数据库的变更记录... 目录python mysql通过Binlog获取变更记录恢复数据1.安装pymysqlreplicat

利用Python编写一个简单的聊天机器人

《利用Python编写一个简单的聊天机器人》这篇文章主要为大家详细介绍了如何利用Python编写一个简单的聊天机器人,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 使用 python 编写一个简单的聊天机器人可以从最基础的逻辑开始,然后逐步加入更复杂的功能。这里我们将先实现一个简单的

基于Python开发电脑定时关机工具

《基于Python开发电脑定时关机工具》这篇文章主要为大家详细介绍了如何基于Python开发一个电脑定时关机工具,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录1. 简介2. 运行效果3. 相关源码1. 简介这个程序就像一个“忠实的管家”,帮你按时关掉电脑,而且全程不需要你多做

Python实现高效地读写大型文件

《Python实现高效地读写大型文件》Python如何读写的是大型文件,有没有什么方法来提高效率呢,这篇文章就来和大家聊聊如何在Python中高效地读写大型文件,需要的可以了解下... 目录一、逐行读取大型文件二、分块读取大型文件三、使用 mmap 模块进行内存映射文件操作(适用于大文件)四、使用 pand

python实现pdf转word和excel的示例代码

《python实现pdf转word和excel的示例代码》本文主要介绍了python实现pdf转word和excel的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价... 目录一、引言二、python编程1,PDF转Word2,PDF转Excel三、前端页面效果展示总结一