如何绘制分段余弦函数并实现周期性？Python代码实现咨询

嘿，我来帮你搞定这个分段余弦函数的实现和周期性问题！先聊聊你现有代码里的几个小问题，再给你一个更简洁高效的实现思路。

你的原代码存在几个可以优化的点：

• 手动循环遍历每个时间点效率太低啦，numpy本来就支持批量处理，完全不用这么麻烦
• 函数的参数设计有点绕，time参数被当成总时长生成t_axis，这样函数只能输出固定长度的数组，不够灵活
• 没有处理周期性的逻辑，目前只覆盖了0到T的区间，没法重复这个波形
• 代码还没写完呢，else块里的ind +=...不完整

接下来咱们用更地道的numpy写法来实现，同时搞定周期性：

import numpy as np
import matplotlib.pyplot as plt

def periodic_piecewise_cos(t, Ts1=0.3, Ts2=0.4):
    # 定义周期长度：这里咱们把0到Ts2作为一个周期，因为之后函数值都是0
    period = Ts2
    # 把所有时间点映射到[0, period)区间，这一步就实现了周期性！
    t_mod = t % period
    
    # 初始化输出数组，默认值为0（对应第三段的情况）
    ft = np.zeros_like(t_mod)
    
    # 处理第一段：0 ≤ t < Ts1
    mask_first = t_mod < Ts1
    ft[mask_first] = np.cos(np.pi * t_mod[mask_first] / Ts1)
    
    # 处理第二段：Ts1 ≤ t < Ts2
    mask_second = (t_mod >= Ts1) & (t_mod < Ts2)
    # 这里沿用你原代码里的表达式，要是和原图像有偏差，可以调整这个公式
    ft[mask_second] = np.cos(np.pi * (t_mod[mask_second] + Ts2 - Ts1) / Ts1)
    
    return ft

# 生成测试时间序列，比如从0到2，覆盖5个周期（因为周期是0.4）
t = np.linspace(0, 2, 1000)
y = periodic_piecewise_cos(t)

# 绘制图像
plt.figure(figsize=(10, 4))
plt.plot(t, y)
plt.xlabel('Time')
plt.ylabel('f(t)')
plt.title('Periodic Piecewise Cosine Function')
plt.grid(True)
plt.show()

关键细节说明： #

• 周期性实现：用t % period把任意时间点映射到一个周期内，不管输入的时间跨度有多大，都会重复0到Ts2的波形
• 向量化操作：用布尔掩码（mask_first、mask_second）批量处理符合条件的时间点，比循环快得多，代码也更简洁
• 灵活性：函数输入是时间数组t，你可以传入任意长度、任意范围的时间序列，输出对应的值

如果你的周期不是Ts2（比如希望0到1秒为一个周期，其中Ts2到1秒都是0），只需要把period改成你想要的长度就行，比如period = 1.0。另外，要是第二段的波形和原图像对不上，你可以检查一下那段的余弦表达式，比如是否需要调整相位或者分母，根据原图像的形状微调就好。

内容的提问来源于stack exchange，提问作者kemal