Python实现正弦波生成GUI图形界面

下面介绍一个通过Python实现正弦波生成的图形界面程序，能够生成多个不同频率，不同幅值的正弦波叠加在一起后的效果并进行输出保存，可以实现如下效果

1、频率10HZ,幅值100，采样率1000，数据长度1000，生成效果如下

二、频率10 和100HZ，幅值均为100，采样率1000，数据长度1000，叠加在一起的样子如下图：

具体的源码如下，自取

import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg


# 生成正弦波并显示
def generate_sine_wave():
    try:
        frequencies = list(map(float, frequency_entry.get().split()))
        amplitudes = list(map(float, amplitude_entry.get().split()))
        data_length = int(length_entry.get())
        sampling_rate = float(sampling_rate_entry.get())

        if len(frequencies) != len(amplitudes):
            raise ValueError("频率和幅度的数量必须相同。")

        t = np.linspace(0, data_length / sampling_rate, data_length)
        global signal
        signal = np.zeros(data_length)

        for freq, amp in zip(frequencies, amplitudes):
            signal += amp * np.sin(2 * np.pi * freq * t)

        fig, ax = plt.subplots()
        ax.plot(t, signal)
        ax.set_title('生成的正弦波')
        ax.set_xlabel('时间 (秒)')
        ax.set_ylabel('幅度')
        ax.grid(True)

        canvas = FigureCanvasTkAgg(fig, master=window)
        canvas.draw()
        canvas.get_tk_widget().grid(row=6, column=0, columnspan=2)

    except Exception as e:
        messagebox.showerror("错误", str(e))


# 保存生成的数据到TXT文件
def save_data():
    try:
        if 'signal' not in globals():
            raise ValueError("请先生成正弦波数据。")

        file_path = filedialog.asksaveasfilename(defaultextension=".txt",
                                                 filetypes=[("文本文件", "*.txt"), ("所有文件", "*.*")])
        if file_path:
            with open(file_path, 'w') as file:
                data_str = ' '.join(map(str, signal))
                file.write(data_str)
            messagebox.showinfo("成功", "数据保存成功")
    except Exception as e:
        messagebox.showerror("错误", str(e))


# 创建主窗口
window = tk.Tk()
window.title("正弦波生成器")

frame = ttk.Frame(window, padding="10")
frame.grid(row=0, column=0, columnspan=2)

ttk.Label(frame, text="输入频率（Hz，用空格分隔）:").grid(row=0, column=0)
frequency_entry = ttk.Entry(frame, width=50)
frequency_entry.grid(row=0, column=1)

ttk.Label(frame, text="输入幅度（用空格分隔）:").grid(row=1, column=0)
amplitude_entry = ttk.Entry(frame, width=50)
amplitude_entry.grid(row=1, column=1)

ttk.Label(frame, text="输入数据长度:").grid(row=2, column=0)
length_entry = ttk.Entry(frame, width=50)
length_entry.grid(row=2, column=1)

ttk.Label(frame, text="输入采样率（Hz）:").grid(row=3, column=0)
sampling_rate_entry = ttk.Entry(frame, width=50)
sampling_rate_entry.grid(row=3, column=1)

ttk.Button(frame, text="生成正弦波", command=generate_sine_wave).grid(row=4, column=0, columnspan=2)
ttk.Button(frame, text="保存数据", command=save_data).grid(row=5, column=0, columnspan=2)

# 运行主窗口
window.mainloop()