Native Argmax

argmax.py

"""
A native implementation of argmax.

https://towardsdatascience.com/there-is-no-argmax-function-for-python-list-cd0659b05e49

Tests::

    pytest --doctest-modules -vvv argmax.py

Linting::

    mypy argmax
"""
from typing import Sequence


def argmax(iterable: Sequence) -> int:
    """
    Return the index of the largest item in an iterable.

    Parameters
    ----------
    iterable : Sequence
        The object in which to find the largest item.

    Examples
    --------
    >>> argmax([3, 31, 29])
    1
    """
    return max(range(len(iterable)), key=lambda i: iterable[i])