Python量化突破:通达信120日天量突破追踪指标主图学习

import pandas as pdimport numpy as npimport matplotlib.pyplot as pltimport matplotlib.patches as mpatchesfrom matplotlib.lines import Line2Ddef convert_tdx(df):    data = df.copy()    data['LIJINZW1'] = (abs(data['close'].shift(1) * 1.1 - data['close']) < 0.01) & (data['high'] == data['close'])    data['LIJINZW2'] = np.nan    for i in range(len(data)):        if i >= 120:            vol_slice = data['volume'].iloc[i-120:i]            if len(vol_slice) > 0:                max_vol_idx = vol_slice.idxmax()                data.loc[data.index[i], 'LIJINZW2'] = data.loc[max_vol_idx, 'high']    data['LIJINZW4'] = data['close'].rolling(window=5).mean()  # MA5    data['LIJINZW5'] = data['close'].rolling(window=10).mean()  # MA10    data['LIJINZW6'] = data['close'].rolling(window=20).mean()  # MA20    data['LIJINZW7'] = data['close'].rolling(window=60).mean()  # MA60    data['LIJINZW8'] = data['close'].rolling(window=120).mean()  # MA120    data['LIJINZW9'] = data['close'].rolling(window=250).mean()  # MA250    data['LIJINZW3'] = (data['LIJINZW2'].shift(1) != data['LIJINZW2']) & data['LIJINZW1']    return data