Переписать функцию в Python - Оптимизировано

Question

Есть ли способ оптимизировать эту функцию? Исходная версия имеет больше параметров внутри кортежей, поэтому она не масштабируется с большим количеством if и elif.

from statistics import median

rule=[1,2,3]
direction=['up','down']

def function_db(mount,rule,direction):

        item=range(1,10)

        # --------------------------------------------------------------------------------
        # Porcentajes
        # --------------------------------------------------------------------------------
        if rule==1 and direction == 'up' :
            final_item = min(item)*(1+mount/100)

        elif rule==2 and direction == 'up' :
            final_item = max(item)*(1+mount/100)

        elif rule==3 and direction == 'up' :
            final_item = median(item)*(1+mount/100)

        elif rule==1 and direction == 'down' :
            final_item = min(item)*(1-mount/100)

        elif rule==2 and direction == 'down' :
            final_item = max(item)*(1-mount/100)

        elif rule==3 and direction == 'down' :
            final_item = median(item)*(1-mount/100)

        return final_item


function_db(10,3,"up")

Answer 1

from statistics import median

rule=[1,2,3]
direction=['up','down']

def function_db(mount,rule,direction):
        dic={1:min,2:max,3:median}
        direct = {'up':1,'down':-1}
        item=range(1,10)
        final_item = dic[rule](item)*(1+(mount/100)*direct[direction])
        return final_item

Answer 2

Я пытался оптимизировать компоновку кода (уменьшив количество if-elses), но все еще сохраняя читабельность. Я предполагаю, что правил будет намного больше, чем указаний, поэтому решил поместить правила в список функций. Таким образом, добавление нового правила включает просто добавление имени функции в rules.

Очевидно, что можно выделить некоторые из функционально-постоянных частей multiplicand, но я подумал, что это сильно пожертвовало читабельностью.

Я также подумал, что стоит проверить (и поднять) ошибки.

from statistics import median

def function_db( mount, rule, direction ):

    item  = range( 1, 10 )
    rules = [ min, max, median ]

    if ( direction == 'up' ):
        multiplicand = ( 1+mount/100 )
    elif ( direction == 'down' ):
        multiplicand = ( 1-mount/100 )
    else:
        raise( ValueError( 'Unknown direction [' + str( direction ) + ']' ) )

    if ( rule == 0 or rule > len( rules ) ):
        raise( ValueError( 'Unknown rule-code [' + str( rule ) + ']' ) )


    # Compute the result
    multiplier = rules[ rule-1 ]( item )

    final_item = multiplier * multiplicand
    return final_item



print( "RESULT=%f" % ( function_db( 10, 3, "up" ) ) )

Answer 3

На самом деле это может быть даже одна строка:

def function_db(mount,rule,direction):
     items = [0, 9, 5]
     directions = {"up": 1, "down": -1}

     return items[rule-1]*(1+ directions[direction]*mount/100)