Ini adalah fungsi ATR sumber terbuka yang ditulis oleh pemilik kumpulan, dan kami meminta anda untuk menulis versi PANDAS tanpa menggunakan fungsi TALIB.
def ATR ((rekod, tempoh=14):
jika len (rekod) == 0:
Kembali []
jika
R = Std._zeros(len(records))
m = 0.0
n = 0.0
for i in xrange(0, len(records)):
TR = 0
if i == 0:
TR = records[i]['High'] - records[i]['Low']
else:
TR = max(records[i]['High'] - records[i]['Low'], abs(records[i]['High'] - records[i - 1]['Close']), abs(records[i - 1]['Close'] - records[i]['Low']))
m += TR
if i < period:
n = m / (i + 1)
else:
n = (((period - 1) * n) + TR) / period
R[i] = n
return R
RumputTidak ada perbezaan.