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 panda库从基础到高级操作分析

《pythonpanda库从基础到高级操作分析》本文介绍了Pandas库的核心功能,包括处理结构化数据的Series和DataFrame数据结构,数据读取、清洗、分组聚合、合并、时间序列分析及大数据... 目录1. Pandas 概述2. 基本操作:数据读取与查看3. 索引操作:精准定位数据4. Group

Python pandas库自学超详细教程

《Pythonpandas库自学超详细教程》文章介绍了Pandas库的基本功能、安装方法及核心操作,涵盖数据导入(CSV/Excel等)、数据结构(Series、DataFrame)、数据清洗、转换... 目录一、什么是Pandas库(1)、Pandas 应用(2)、Pandas 功能(3)、数据结构二、安

Python使用Tenacity一行代码实现自动重试详解

《Python使用Tenacity一行代码实现自动重试详解》tenacity是一个专为Python设计的通用重试库,它的核心理念就是用简单、清晰的方式,为任何可能失败的操作添加重试能力,下面我们就来看... 目录一切始于一个简单的 API 调用Tenacity 入门:一行代码实现优雅重试精细控制:让重试按我

Python安装Pandas库的两种方法

《Python安装Pandas库的两种方法》本文介绍了三种安装PythonPandas库的方法,通过cmd命令行安装并解决版本冲突,手动下载whl文件安装,更换国内镜像源加速下载,最后建议用pipli... 目录方法一:cmd命令行执行pip install pandas方法二:找到pandas下载库,然后

Python实现网格交易策略的过程

《Python实现网格交易策略的过程》本文讲解Python网格交易策略,利用ccxt获取加密货币数据及backtrader回测,通过设定网格节点,低买高卖获利,适合震荡行情,下面跟我一起看看我们的第一... 网格交易是一种经典的量化交易策略,其核心思想是在价格上下预设多个“网格”,当价格触发特定网格时执行买

Python标准库之数据压缩和存档的应用详解

《Python标准库之数据压缩和存档的应用详解》在数据处理与存储领域,压缩和存档是提升效率的关键技术,Python标准库提供了一套完整的工具链,下面小编就来和大家简单介绍一下吧... 目录一、核心模块架构与设计哲学二、关键模块深度解析1.tarfile:专业级归档工具2.zipfile:跨平台归档首选3.

使用Python构建智能BAT文件生成器的完美解决方案

《使用Python构建智能BAT文件生成器的完美解决方案》这篇文章主要为大家详细介绍了如何使用wxPython构建一个智能的BAT文件生成器,它不仅能够为Python脚本生成启动脚本,还提供了完整的文... 目录引言运行效果图项目背景与需求分析核心需求技术选型核心功能实现1. 数据库设计2. 界面布局设计3

Python进行JSON和Excel文件转换处理指南

《Python进行JSON和Excel文件转换处理指南》在数据交换与系统集成中,JSON与Excel是两种极为常见的数据格式,本文将介绍如何使用Python实现将JSON转换为格式化的Excel文件,... 目录将 jsON 导入为格式化 Excel将 Excel 导出为结构化 JSON处理嵌套 JSON:

Python操作PDF文档的主流库使用指南

《Python操作PDF文档的主流库使用指南》PDF因其跨平台、格式固定的特性成为文档交换的标准,然而,由于其复杂的内部结构,程序化操作PDF一直是个挑战,本文主要为大家整理了Python操作PD... 目录一、 基础操作1.PyPDF2 (及其继任者 pypdf)2.PyMuPDF / fitz3.Fre

python设置环境变量路径实现过程

《python设置环境变量路径实现过程》本文介绍设置Python路径的多种方法:临时设置(Windows用`set`,Linux/macOS用`export`)、永久设置(系统属性或shell配置文件... 目录设置python路径的方法临时设置环境变量(适用于当前会话)永久设置环境变量(Windows系统