Как получить результат условия имея состояния?

Question

[Олег Петров]

У меня есть словарь, который содержит тикеры: данные
tickers = {'BTC': [200, 149, 98, 44], 'ETH': [200, 320, 405, 460]}
Я написал функцию, которая может узнавать состояние из условия.
Всего 4 различных состояния
prev и now - ставнение 2 элементов в листе друг с другом. Сегодня и вчера
Например для BTC:
149 и 200
98 и 149
44 и 98

# check state . Описываем состояния для условия. Может быть 4 различных
def check_state(now, prev):
    state = None
    if now >= prev:
        if now <= 1.5 * prev:
                state = 0
    if now >= 1.5 * prev:
        state = 1
    if now < prev:
        if now * 1.5 >= prev:
                state = 2
    if now * 1.5 < prev:
        state = 3
    # возвращаем состояние
    return state

Теперь я хочу прогнать по всем элементам словаря и узнать состояние в каждой точке и на выходе получить такой же словарь состояний. Например Должно получиться так
tickers_state = {'BTC': [None, 3, 3, 3], 'ETH': [None, 1, 0, 0]}
У первого элемента нет состояния, так как у него нет предыдущего значения.
Как грамотно сделать?
P.S. Я:
1)Итерирую все элементы в tickers
2)Итерирую все элементы в list
3)Вызываю функцию check_state и получаю state
Вопросы:
Как исключить первый элемент, так как у него нет предыдущего и состояние None
Что подавать в качестве параметра функции check_state ?

Comments

[NaName]

Олег Петров, есть паттерн "состояние". может поможет упростить код. а может нет, кто знает)

[Олег Петров]

NaName, Ну я вроде как его сам сделал. А что за паттерн?

Answer 1

[Вадим Шаталов]

def list_state(input_list):
    check0 = lambda prev, now: 0 if now <= 1.5 * prev and now >= prev else None
    check1 = lambda prev, nov: 1 if now >= 1.5 * prev else None
    check2 = lambda prev, now: 2 if now * 1.5 >= prev and now < prev else None
    check3 = lambda prev, now: 3 if now * 1.5 < prev else None

    output = [None]
    for index in range(len(input_list)):
        if index > len(input_list)-2:
            break
        prev, now = input_list[index], input_list[index+1]
        for condition in (check0, check1, check2, check3):
            result = condition(prev, now)
            if result is not None:
                output.append(result)
                break
    return output

a =  [200, 320, 405, 460]
print(list_state(a))
# [None, 1, 0, 0]
a = [200, 149, 98, 44]
print(list_state(a))
# [None, 2, 3, 3]

tickers = {'BTC': [200, 149, 98, 44], 'ETH': [200, 320, 405, 460]}
tickers_state = {}
for key, data_list in tickers.items():
    tickers_state[key] = list_state(data_list)
print(tickers_state)
# {'BTC': [None, 2, 3, 3], 'ETH': [None, 1, 0, 0]}

Comments

[Олег Петров]

Спасибо большое.

[Олег Петров]

Привожу мой код. Дело в том, что у меня таких листов будет очень много и они в списке, надо их как-то итерировать.

volume_BTC = [1, 2, 3, 4]
price_BTC = [50, 100, 150, 200]
volume_ETH = [1, 5, 8, 12]
price_ETH = [50, 40, 25, 9]
tickers = [volume_BTC, price_BTC, volume_ETH, price_ETH]
def check_state(now, prev):
    if prev is None:
        return None
    elif now is None:
        return None
    elif now >= prev and now <= 1.5 * prev:
                state = 0
    elif now >= 1.5 * prev:
        state = 1
    elif now < prev and now * 1.5 >= prev:
                state = 2
    elif now * 1.5 < prev:
        state = 3
    # возвращаем состояние
    return state

# Узнаем по каком тикеру сработало условие?
def check_condition(tickers):
    list_tickers =[]
    valueprev = None
    for i_ticker in tickers:
        for valuenow in i_ticker:
            check_this_state = check_state(valuenow, valueprev)
            valueprev = valuenow
            list_tickers.append(check_this_state)
        valueprev = None
    return list_tickers

res = check_condition(tickers)
print(res)

[Вадим Шаталов]

volume_BTC = [1, 2, 3, 4]
price_BTC = [50, 100, 150, 200]
volume_ETH = [1, 5, 8, 12]
price_ETH = [50, 40, 25, 9]
tickers = [volume_BTC, price_BTC, volume_ETH, price_ETH]
output = [list_state(ticker) for ticker in tickers]

Answer 2

[Swartalf]

1. if переделать на if-elif.
2. вложеные if можжно объединить AND 'ом
3 . если после условий логики не будет, то можете сразу делать return
4. Проверять будете циклом? Если да, то для нулевого элемента списка в prev передавайте None

if prev  is None:
  return None

Comments

[Олег Петров]

Все исправил как вы сказали, спасибо, но сразу Return после каждого elif почему-то не получается сделать, пишет ошибка

def check_state(now, prev):
    if prev is None:
        return None
    elif now is None:
        return None
    elif now >= prev and now <= 1.5 * prev:
                state = 0
    elif now >= 1.5 * prev:
        state = 1
    elif now < prev and now * 1.5 >= prev:
                state = 2
    elif now * 1.5 < prev:
        state = 3
    # возвращаем состояние
    return state

Так а что можете посоветовать по тому, чтобы прогнать условия через цикл и получить состояния?
Я:
1)Итерирую все элементы в tickers
2)Итерирую все элементы в list
3)Вызываю функцию check_state и получаю state
Вопросы:
Как исключить первый элемент, так как у него нет предыдущего и состояние None
Что подавать в качестве параметра функции check_state ?

[Swartalf]

Хм. пока писал комментарий изменились условия.
потому 4-й пункт вычеркиваем

UPD.
По поводу итераций нужно понимать какие объемы данных нужно будет обработать.
* Если в листах будет немного значений, то можно переделать фунцию что бы она принимала на вход массив. И тогда можно будет делать итерацию по ключам в tickers, а листы передавать в функцию. По итогу возвращать один список.
* если значений планируется много, то можно в два цикла, итерация по ключам tickers и внутри по элементам листа и собирать результирующий массив в верхнем цикле.

[Олег Петров]

Swartalf, Значений будет очень много. Вот мой полный код теперь

volume_BTC = [1, 2, 3, 4]
price_BTC = [50, 100, 150, 200]
volume_ETH = [1, 5, 8, 12]
price_ETH = [50, 40, 25, 9]
tickers = [volume_BTC, price_BTC, volume_ETH, price_ETH]
tickers_state = {}


# check state . Описываем состояния для первого условия. Может быть 4 различных
def check_state(now, prev):
    if prev is None:
        return None
    elif now is None:
        return None
    # [0; 1.5x]
    elif now >= prev and now <= 1.5 * prev:
                state = 0
    # [1.5x;Бесконечность]
    elif now >= 1.5 * prev:
        state = 1
    # [-1.5x;0]
    elif now < prev and now * 1.5 >= prev:
                state = 2
    # [-Бесконечность]
    elif now * 1.5 < prev:
        state = 3
    # возвращаем состояние
    return state

# Узнаем состояние по первому условию для всего значений в колонке
def check_condition(ticker):
    list_tickers =[]
    valueprev = None
    #for i_ticker in tickers:
    for valuenow in ticker:
        check_this_state = check_state(valuenow, valueprev)
        valueprev = valuenow
        list_tickers.append(check_this_state)
    return list_tickers


for i_ticker in tickers:
    res = check_condition(i_ticker)
    print(res)

Оцените, если не трудно.
Не очень понял, а почему 4й пункт отменяется? Мне как раз понравился ваш совет с None.
Я еще добавил и проверку текущего элемента на None

[Swartalf]

Выше уже дали готовое решение с лямбдами. Имхо, закончите свой вариант и попробуйте оба варианта в работе и скорости.

[Олег Петров]

Swartalf, Да, я как раз закончил свое, когда предложили тот вариант, в любом случае спасибо за помощь и исправления кода.