Создайте функцию в Python, чтобы найти самый высокий из всех аргументов функции, и вернуть «тег» значения

Question

Учтите следующее: р1 = 1; р2 = 5; р3 = 7; высокий = макс (р1, р2, р3).

Функция max вернула бы 7. Я ищу, чтобы создать аналогичную функцию, которая вернула бы "p3". Я создал небольшую функцию (путем простого сравнения) для приведенного выше примера, показанного ниже. однако у меня возникают проблемы, когда количество аргументов возрастает.

def highest(p1,p2,p3):
    if (p1>p2) and (p1>p3):
        return "p1"
    if (p2>p1) and (p2>p3):
        return "p2"
    if (p3>p1) and (p3>p1):
        return "p3"

Есть ли более простой способ сделать это>

Answer 1

Обновление: Пол Хэнкин указал, что max () взял ключевую функцию, которую я не знал.Итак:

>>> def argmax(**kw):
...   return max(kw, key=kw.get)
... 
>>> argmax(foo=3, bar=5, frotz=1, kaka=-3)
'bar'

---

Другие решения для полноты:

В Python 2.7 и 3.x вы можете использовать словарные понимания.

>>> def argmax(**kw):
...     wk = {v:k for k,v in kw.items()}
...     return wk[max(wk)]
... 
>>> argmax(foo=3, bar=5, frotz=1, kaka=-3)   
'bar'

* 1009аккуратный.:)

В более ранних версиях Python вы можете сделать это:

>>> def argmax(**kw):
...     wk = dict([(v,k) for k,v in kw.items()])
...     return wk[max(wk)]
... 
>>> argmax(foo=3, bar=5, frotz=1, kaka=-3)   
'bar'

, который будет работать во всем после Python 2.2 или около того.

Answer 2

Невозможно получить имя переменной , которая имеет самое высокое значение в вызывающей стороне (поскольку это может быть число или сложное выражение), но с помощью Исключительно ключевые аргументы, вы можете получить имя параметра . Примерно так:

def argmax(**kwargs):
    mx = -1e+400 # overflows to -Inf
    amx = None
    for k, v in kwargs.iteritems():
        if v > mx:
            mx = v
            amx = k
    return amx

Работает так:

>>> argmax(a=1,b=2,c=3)
'c'

но выгода в том, что он не работает, если какой-либо из аргументов является позиционным:

>>> argmax(1,2,3)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: argmax() takes exactly 0 arguments (3 given)

В зависимости от того, что вы делаете, эта конструкция может быть более полезной: (с подсказкой http://lemire.me/blog/archives/2008/12/17/fast-argmax-in-python/)

>>> a = [9,99,999]
>>> a.index(max(a))
2

Answer 3

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

def highest(x):
    return 'p' + repr(x.index(max(x))+1)

Answer 4

Очевидно, что он не обрабатывает аргументы переменной длины. Если вы хотите аргумент переменной длины, то это другая проблема. Если у вас есть 10 аргументов, просто добавьте их в определение, и оно вернет правильное имя аргумента (необязательно начинающееся с 'p') Но подвох - это количество аргументов (3, 5, 10 или что-то еще) не является переменной. Вам нужно знать, сколько аргументов вам требуется.

def highest(p1,p2,p3,p4,p5):
    d = locals()
    keys = d.keys()
    max_key = keys[0]
    max_val = d[max_key]
    for i in range(1,len(keys)):
        key = keys[i]
        val = d[key]
        if val > max_val:
            max_val = val
            max_key = key

    return max_key

print highest(3,2,5,10,1)
print highest(1,5,2,2,3)
print highest(5,2,5,1,11)
print highest(3,2,1,1,2)