如何将PineScript代码改写为Python以实现相同价格极值追踪

我帮你分析下问题所在，然后给出和TradingView输出完全一致的实现方案。

首先先明确原PineScript代码的核心逻辑：

src = input.source(close, "Source")
volStop(src) =>
    var max     = src
    var min     = src
    max         := math.max(max, src)
    min         := math.min(min, src)
    [max, min]

[max, min] = volStop(src)
plot(max, "Max", style=plot.style_cross)

这里的var关键字是关键——它会在第一个K线（bar）时初始化一次max和min为当时的src值，之后每一根新K线，都会用上一根K线的max和当前src取较大值更新max，min同理取较小值。本质上就是计算从第一根K线到当前K线的累积全局最大值/最小值。

你写的Python代码问题出在两个地方：

• 用shift(1).fillna(0)把第一个K线的前置值设为0，这和PineScript中第一个K线的max/min就是第一个收盘价的逻辑完全不符
• 循环里通过shift(1)获取的是上一行原始的max值，而不是更新后的累积值，导致逻辑走偏

下面给你两种正确的实现方式：

方法一：贴近PineScript逐bar执行的循环写法 #

这种写法完全复刻PineScript的执行逻辑，适合理解过程：

import pandas as pd

# 假设你的价格数据存在price_df中，'Close'是目标列
name_var = 'Close'
maxmin = pd.DataFrame(index=price_df.index)

# 初始化第一个值，对应PineScript的var初始化逻辑
current_max = price_df[name_var].iloc[0]
current_min = price_df[name_var].iloc[0]

# 先填充第一行的值
maxmin.loc[price_df.index[0], 'max'] = current_max
maxmin.loc[price_df.index[0], 'min'] = current_min

# 从第二行开始逐行更新
for i in range(1, len(price_df)):
    current_price = price_df[name_var].iloc[i]
    # 更新累积最大值/最小值
    current_max = max(current_max, current_price)
    current_min = min(current_min, current_price)
    # 写入结果
    maxmin.loc[price_df.index[i], 'max'] = current_max
    maxmin.loc[price_df.index[i], 'min'] = current_min

方法二：用Pandas内置函数的简洁写法（推荐） #

Pandas已经内置了累积最大/最小的函数，一行代码就能实现，效率比循环高很多：

import pandas as pd

name_var = 'Close'
maxmin = pd.DataFrame()
# cummax()计算从第一个元素到当前元素的累积最大值，完全匹配PineScript逻辑
maxmin['max'] = price_df[name_var].cummax()
# cummin()计算从第一个元素到当前元素的累积最小值
maxmin['min'] = price_df[name_var].cummin()

这两种方法的输出都会和TradingView上的结果完全一致，你可以任选一种使用。

内容的提问来源于stack exchange，提问作者Jakub Szurlej