Bolehkah anda mengubah kod ini menjadi kod PANDAS?

Penulis:Syue, Dicipta: 2021-06-29 21:03:55, Dikemas kini:

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 Close tidak dalam rekod[0]: mengangkat TA.ATR hujah mesti KLine

    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

Lebih lanjut

RumputTidak ada perbezaan.