绘制李萨如图,数据保存excel,txt

前置

看到网络上绘制的这么漂亮的数学图形,觉得能够用一个代码绘制全部的这些图形,就能提高不少效率,所以于是就心血来潮:

这是一个绘制李萨如图并保存数据的Python代码示例。下面是每行注释的详细解释:

python 复制代码
import os  # 导入操作系统模块
from datetime import datetime  # 导入日期时间模块
import numpy as np  # 导入NumPy库
import matplotlib.pyplot as plt  # 导入Matplotlib库
import pandas as pd  # 导入Pandas库

# 创建文件夹函数
def create_directory(directory):
    if not os.path.exists(directory):  # 如果目录不存在
        os.makedirs(directory)  # 创建目录

# 检索可用英语字体函数
def search_english_font(directory):
    font_files = os.listdir(directory)  # 获取目录下的文件列表
    for file in font_files:
        if file.endswith(".ttf"):  # 如果文件以".ttf"结尾
            return os.path.join(directory, file)  # 返回字体文件的完整路径
    return None  # 如果未找到可用的英语字体文件,返回None

# 绘制李萨如图函数
def plot_lissajous(a, b, delta):
    t = np.linspace(0, 2*np.pi, 1000)  # 在0到2*pi之间生成1000个点
    x = np.sin(a*t + delta)  # 计算x坐标
    y = np.sin(b*t)  # 计算y坐标
    fig, ax = plt.subplots()  # 创建图形和坐标轴对象
    ax.plot(x, y)  # 绘制曲线
    ax.axis("equal")  # 设置坐标轴刻度相等
    ax.set_title("Lissajous Curve (a={}, b={}, delta={})".format(a, b, delta))  # 设置标题
    return fig, ax  # 返回图形和坐标轴对象

# 保存数据到excel和txt文件函数
def save_data(data, filename):
    excel_path = os.path.join(directory_path, "代码print相关/结构图", filename + ".xlsx")  # 构造Excel文件路径
    data.to_excel(excel_path, index=False)  # 将数据保存到Excel文件
    print("数据已保存到:" + excel_path)  # 打印保存成功的消息

    txt_path = os.path.join(directory_path, "代码print相关/结构图", filename + ".txt")  # 构造txt文件路径
    data.to_csv(txt_path, sep="\t", index=False)  # 将数据保存到txt文件
    print("数据已保存到:" + txt_path)  # 打印保存成功的消息

# 设置目录路径和文件名前缀
directory_path = "/storage/emulated/0/文件/"
prefix = "编号.库样式."

# 构造当前时间戳
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")

# 组合文件名前缀
filename_prefix = prefix + "Lissajous."

# 组合文件名
filename = filename_prefix + timestamp

# 构造完整文件路径
file_path = os.path.join(directory_path, "代码print相关/结构图", filename)

# 创建目录
create_directory(os.path.dirname(file_path))

# 检查中文字体是否存在
chinese_font_path = "/storage/emulated/0/文件/字体大全/字体包/宋体.ttf"
if not os.path.exists(chinese_font_path):  # 如果中文字体文件不存在
    chinese_font_path = "/storage/emulated/0/文件/字体大全/字体包/"  # 设置备用路径

# 检索可用英语字体
english_font_directory = "/storage/emulated/0/文件/字体大全/"
english_font_path = search_english_font(english_font_directory)

# 打印注释
print("这个代码文件使用了以下字体:")
print("- 中文字体:" + chinese_font_path)
if english_font_path:
    print("- 英文字体:" + english_font_path)

# 绘制李萨如图并保存数据
a = 3
b = 4
deltas = np.linspace(0, 2*np.pi, 9)[:-1]  # 在0到2*pi之间生成8个delta值
for i, delta in enumerate(deltas):
    fig, ax = plot_lissajous(a, b, delta)  # 绘制李萨如图
    data = pd.DataFrame({"x": ax.lines[0].get_xdata(), "y": ax.lines[0].get_ydata()})  # 获取曲线数据
    save_data(data, filename + "." + str(i+1))  # 保存数据到文件

plt.show()  # 显示绘制的图形

这个代码绘制李萨如图,并将数据保存到Excel和txt文件中。每行有趣的注释都描述了代码的功能和作用,这样就能更好地理解代码的逻辑和实现。
例中仅保存数据到Excel和txt文件,并未包含保存图片的部分。若保存图片文件,可以添加以下代码来保存图片:

# 组合图片文件名
image_filename = filename_prefix + "Lissajous_" + str(i+1) + ".png"

# 构造图片文件路径
image_path = os.path.join(directory_path, "代码print相关/结构图", image_filename)

# 保存图片
fig.savefig(image_path)
print("图片已保存到:" + image_path)

输出结果为:

相关推荐
Leyla12 分钟前
【代码重构】好的重构与坏的重构
前端
今天也要加油丫14 分钟前
`re.compile(r“(<.*?>)“)` 如何有效地从给定字符串中提取出所有符合 `<...>` 格式的引用
python
影子落人间16 分钟前
已解决npm ERR! request to https://registry.npm.taobao.org/@vant%2farea-data failed
前端·npm·node.js
世俗ˊ40 分钟前
CSS入门笔记
前端·css·笔记
子非鱼92140 分钟前
【前端】ES6:Set与Map
前端·javascript·es6
6230_1 小时前
git使用“保姆级”教程1——简介及配置项设置
前端·git·学习·html·web3·学习方法·改行学it
想退休的搬砖人1 小时前
vue选项式写法项目案例(购物车)
前端·javascript·vue.js
加勒比海涛1 小时前
HTML 揭秘:HTML 编码快速入门
前端·html
啥子花道1 小时前
Vue3.4 中 v-model 双向数据绑定新玩法详解
前端·javascript·vue.js
麒麟而非淇淋1 小时前
AJAX 入门 day3
前端·javascript·ajax