请用python生成计算最大回撤的代码

以下是一个简单的Python代码示例，用于计算给定数据集的最大回撤：

import numpy as np

def max_drawdown(data):
    """
    计算最大回撤
    data: 输入数据，可以是价格序列等
    """

    max_return = np.maximum.accumulate(data)  # 计算每个时间点之前的最大值
    drawdown = (data - max_return) / max_return  # 计算回撤率
    max_drawdown = np.min(drawdown)  # 计算最大回撤

    return max_drawdown

这里使用了NumPy库中的函数来计算每个时间点之前的最大值和回撤率。最大回撤是回撤率的最小值。请注意，这个函数只计算最大回撤，而不是最大回撤的时间段。如果需要计算时间段，请参考其他方法。