利用公式编写ATR指标代码

以下是一个示例代码，用于计算ATR指标：

import pandas as pd

def calculate_atr(data, n):
    atr = pd.Series(index=data.index)
    atr[0] = data['high'][0] - data['low'][0]
    
    for i in range(1, len(data)):
        tr = max(data['high'][i] - data['low'][i], abs(data['high'][i] - data['close'][i-1]), abs(data['low'][i] - data['close'][i-1]))
        atr[i] = (atr[i-1] * (n-1) + tr) / n
    
    return atr

这个函数接受一个包含高、低和收盘价的数据框，并返回一个包含ATR指标值的序列。参数n表示计算ATR的周期。

请注意，此代码仅用于示例目的，并未经过完整测试，可能还需要根据实际情况进行调整和优化