Pandas, Python. Есть отрывок кода индикатора RSI с использованием pandas, есть список, куда его притулить?

Question

[Aurum_Stella]

Есть код взят на просторах интернета, суть его в том "если я правильно понял" что он анализирует список чисел и выводит значение RSI (индекс относительной силы).
Сначала я два дня пытался самостоятельно написать этот индикатор по формулам с интернета, но постоянно получал не тот результат сравнивал на сайте tradingview. Наткнулся тут на код который вроде как выполняет требуемые мне действия, но я не знаю Pandas.
Суть такова, у меня есть список чисел [1,2,3...n], он хранится в json файле, как мне передать этот список в данный код? Куда притулить переменную с списком что бы он заработал?

Приму любую помощь, и буду очень благодарен, впредь обещаю подучить pandas.

spoiler

import pandas as pd
import numpy as np


def rsi_tradingview(ohlc: pd.DataFrame, period: int = 14, round_rsi: bool = True):
    """ Implements the RSI indicator as defined by TradingView on March 15, 2021.
        The TradingView code is as follows:
        //@version=4
        study(title="Relative Strength Index", shorttitle="RSI", format=format.price, precision=2, resolution="")
        len = input(14, minval=1, title="Length")
        src = input(close, "Source", type = input.source)
        up = rma(max(change(src), 0), len)
        down = rma(-min(change(src), 0), len)
        rsi = down == 0 ? 100 : up == 0 ? 0 : 100 - (100 / (1 + up / down))
        plot(rsi, "RSI", color=#8E1599)
        band1 = hline(70, "Upper Band", color=#C0C0C0)
        band0 = hline(30, "Lower Band", color=#C0C0C0)
        fill(band1, band0, color=#9915FF, transp=90, title="Background")
    :param ohlc:
    :param period:
    :param round_rsi:
    :return: an array with the RSI indicator values
    """

    delta = ohlc["close"].diff()

    up = delta.copy()
    up[up < 0] = 0
    up = pd.Series.ewm(up, alpha=1/period).mean()

    down = delta.copy()
    down[down > 0] = 0
    down *= -1
    down = pd.Series.ewm(down, alpha=1/period).mean()

    rsi = np.where(up == 0, 0, np.where(down == 0, 100, 100 - (100 / (1 + up / down))))

    return np.round(rsi, 2) if round_rsi else rsi

Comments

[Алан Гибизов]

Не вдаваясь в математику, набросал пример для работы с приведенной функцией. Не уверен, что оно выдает то, что ожидается, но что-то оно таки выдает.

код примера

import json

import numpy as np
import pandas as pd

from pathlib import Path

#  тут я формирую тестовый файл json со списком длиной 50
n = 50
list_for_json = [i for i in range(n)]
my_json_file = Path('myjson.json')
with open(my_json_file, 'w') as myfile:
    json.dump(fp=myfile, obj=list_for_json)

#  тут открываю этот файл, создаю из него dataframe Pandas и именую колонку с данными 'close', 
#  так как такое имя используется в приведенной вами функции.
with open(my_json_file, 'r') as myfile:
    df = pd.read_json(myfile)
df.rename(columns={0: 'close'}, inplace=True)
print(df.head())

#  собственно функция без изменений
def rsi_tradingview(ohlc: pd.DataFrame, period: int = 14, round_rsi: bool = True):
    """ Implements the RSI indicator as defined by TradingView on March 15, 2021.
        The TradingView code is as follows:
        //@version=4
        study(title="Relative Strength Index", shorttitle="RSI", format=format.price, precision=2, resolution="")
        len = input(14, minval=1, title="Length")
        src = input(close, "Source", type = input.source)
        up = rma(max(change(src), 0), len)
        down = rma(-min(change(src), 0), len)
        rsi = down == 0 ? 100 : up == 0 ? 0 : 100 - (100 / (1 + up / down))
        plot(rsi, "RSI", color=#8E1599)
        band1 = hline(70, "Upper Band", color=#C0C0C0)
        band0 = hline(30, "Lower Band", color=#C0C0C0)
        fill(band1, band0, color=#9915FF, transp=90, title="Background")
    :param ohlc:
    :param period:
    :param round_rsi:
    :return: an array with the RSI indicator values
    """

    delta = ohlc["close"].diff()

    up = delta.copy()
    up[up < 0] = 0
    up = pd.Series.ewm(up, alpha=1/period).mean()

    down = delta.copy()
    down[down > 0] = 0
    down *= -1
    down = pd.Series.ewm(down, alpha=1/period).mean()

    rsi = np.where(up == 0, 0, np.where(down == 0, 100, 100 - (100 / (1 + up / down))))

    return np.round(rsi, 2) if round_rsi else rsi

#  ну а дальше просто вызываю функцию и передаю в неё dataframe и печатаю, что она возвращает. #  Насколько это вменяемо внутри функции, не знаю.

print(rsi_tradingview(ohlc=df))

[Aurum_Stella]

Алан Гибизов, Спасибо! Помогло еще и как, правда вначале функция выводила не совсем то что нужно. Правильный вывело когда прямо в конце функции прописал print(np.round(rsi, 2)), оно сразу выдает нужный результат, только нужно подавать намного больше входящих данных. Я подал за 200 периодов, и оно выдало точный результат.

Спасибо огромное!

Дополняю, от радости не увидел что у меня функция ни куда ничего не выводит.
Принт внутри функции не нужен)

Answer 1

[Алан Гибизов]

Не вдаваясь в математику, набросал пример для работы с приведенной функцией. Не уверен, что оно выдает то, что ожидается, но что-то оно таки выдает.

код примера

import json

import numpy as np
import pandas as pd

from pathlib import Path

#  тут я формирую тестовый файл json со списком длиной 50
n = 50
list_for_json = [i for i in range(n)]
my_json_file = Path('myjson.json')
with open(my_json_file, 'w') as myfile:
    json.dump(fp=myfile, obj=list_for_json)

#  тут открываю этот файл, создаю из него dataframe Pandas и именую колонку с данными 'close', 
#  так как такое имя используется в приведенной вами функции.
with open(my_json_file, 'r') as myfile:
    df = pd.read_json(myfile)
df.rename(columns={0: 'close'}, inplace=True)
print(df.head())

#  собственно функция без изменений
def rsi_tradingview(ohlc: pd.DataFrame, period: int = 14, round_rsi: bool = True):
    """ Implements the RSI indicator as defined by TradingView on March 15, 2021.
        The TradingView code is as follows:
        //@version=4
        study(title="Relative Strength Index", shorttitle="RSI", format=format.price, precision=2, resolution="")
        len = input(14, minval=1, title="Length")
        src = input(close, "Source", type = input.source)
        up = rma(max(change(src), 0), len)
        down = rma(-min(change(src), 0), len)
        rsi = down == 0 ? 100 : up == 0 ? 0 : 100 - (100 / (1 + up / down))
        plot(rsi, "RSI", color=#8E1599)
        band1 = hline(70, "Upper Band", color=#C0C0C0)
        band0 = hline(30, "Lower Band", color=#C0C0C0)
        fill(band1, band0, color=#9915FF, transp=90, title="Background")
    :param ohlc:
    :param period:
    :param round_rsi:
    :return: an array with the RSI indicator values
    """

    delta = ohlc["close"].diff()

    up = delta.copy()
    up[up < 0] = 0
    up = pd.Series.ewm(up, alpha=1/period).mean()

    down = delta.copy()
    down[down > 0] = 0
    down *= -1
    down = pd.Series.ewm(down, alpha=1/period).mean()

    rsi = np.where(up == 0, 0, np.where(down == 0, 100, 100 - (100 / (1 + up / down))))

    return np.round(rsi, 2) if round_rsi else rsi

#  ну а дальше просто вызываю функцию и передаю в неё dataframe и печатаю, что она возвращает. #  Насколько это вменяемо внутри функции, не знаю.

print(rsi_tradingview(ohlc=df))

Comments

[mkone112]

На первый взгляд, в совмещении Path и open - нет смысла.

[Алан Гибизов]

mkone112, виноват - мне нравится Path своим механизмом, удобством и простотой; и я его стараюсь применять всюду; а в примерах на тостере ещё и для того, чтобы начинающие обратили на него внимание. Маленькая реклама :))

[mkone112]

Алан Гибизов, основная проблема Path - он поддерживается далеко не везде, и это походу надолго.

[Алан Гибизов]

mkone112, кстати, там где не поддерживается pathlib, порой срабатывает pathlib2. Так что костыль придумать можно… но да, это печально.