python用一个分段函数描述三角波

下面是一个用 Python 描述三角波的分段函数：

import math

def triangle_wave(x):
    x = x - math.floor(x) # 将 x 转换为 [0, 1) 范围内的数
    if x < 0.25:
        return 4 * x
    elif x < 0.75:
        return 2 - 4 * x
    else:
        return -4 + 4 * x

这个函数将输入的 x 值转换为 [0, 1) 范围内的数，然后根据 x 的值返回相应的三角波函数值。