stock/Chan241206/validate_fractals.py

def identify_fractals(df):
    """
    根据基本定义识别初步的顶底分型。
    不进行有效性判定（如间隔、极值强化、交替性等）。

    分型定义（3K线窗口）:
    - 顶分型(Top): 中间K线的High大于相邻两K线的High
    - 底分型(Bottom): 中间K线的Low小于相邻两K线的Low

    :param df: 已经去除包含关系的K线DataFrame，要求有 'High'、'Low' 列
    :return: 分型列表 [(index, 'Top'), (index, 'Bottom')]，按时间顺序列出
    """
    fractals = []
    # 遍历从第1根K线到倒数第2根K线（i-1、i、i+1可用）
    for i in range(1, len(df) - 1):
        # 顶分型判断
        if df.loc[i, 'High'] > df.loc[i - 1, 'High'] and df.loc[i, 'High'] > df.loc[i + 1, 'High']:
            fractals.append((i, 'Top'))

        # 底分型判断
        elif df.loc[i, 'Low'] < df.loc[i - 1, 'Low'] and df.loc[i, 'Low'] < df.loc[i + 1, 'Low']:
            fractals.append((i, 'Bottom'))

    return fractals