Funciones en Python: cómo llamar y escribir funciones

Argumentos por defecto

Son aquellos que toman un valor predeterminado si no se pasa ningún valor durante la llamada. Puedes asignar ese valor con el operador =, como en este ejemplo:

Argumentos obligatorios

Como su nombre sugiere, son los que tienen que estar. Deben pasarse durante la llamada y en el orden correcto, como en el ejemplo:

Necesitas argumentos que se asignen a los parámetros a y b para llamar a la función sin errores. Si intercambias a y b, el resultado no cambiará, pero sí podría hacerlo si modificas plus() así:

Argumentos por palabra clave

Si quieres asegurarte de que llamas a los parámetros en el orden correcto, puedes usar argumentos por palabra clave en la llamada. Sirven para identificar los valores por el nombre del parámetro. Tomemos el ejemplo anterior para verlo más claro:

Observa que, usando argumentos por palabra clave, también puedes cambiar el orden de los parámetros y seguir obteniendo el mismo resultado al ejecutar la función:

Número variable de argumentos

Si no sabes cuántos argumentos vas a pasar a una función, puedes usar la siguiente sintaxis con *args:

El asterisco (*) va antes del nombre de la variable que almacenará los valores de todos los argumentos variables sin palabra clave. Fíjate en que también podrías haber pasado *varint, *var_int_args o cualquier otro nombre a la función plus().

Truco: prueba a sustituir *args por otro nombre que incluya el asterisco. ¡Verás que el código sigue funcionando!

Verás que la función anterior usa la función integrada sum() de Python para sumar todos los argumentos que se pasan a plus().

Variables globales vs. locales

En general, las variables definidas dentro del cuerpo de una función tienen alcance local y las que se definen fuera tienen alcance global. Eso significa que las variables locales se definen dentro de un bloque de función y solo se pueden usar dentro de esa función, mientras que las globales pueden ser accedidas por todas las funciones de tu script:

Verás que obtendrás un NameError que dice que name 'total' is not defined cuando intentes imprimir la variable local total, definida dentro del cuerpo de la función. La variable init, en cambio, se puede imprimir sin problemas.

Parámetros solo posicionales y solo por palabra clave

Desde Python 3.8, puedes controlar mejor cómo deben pasarse los argumentos usando dos marcadores especiales en la firma de tu función: / y *. Todo lo que está antes de / solo puede pasarse posicionalmente; todo lo que está después de * solo puede pasarse por palabra clave.

def greet(name, /, greeting="Hello", *, punctuation="!"):
    print(f"{greeting} {name}{punctuation}")

 Así se ve al llamar a la función:

greet("Alice")                          # funciona
greet("Alice", greeting="Hi")           # funciona
greet("Alice", "Hi", punctuation="?")   # funciona

greet(name="Alice")                     # TypeError: name es solo posicional
greet("Alice", "Hi", "?")               # TypeError: punctuation es solo por palabra clave

¿Por qué querrías esto? Dos razones principales.

Primero, los parámetros solo posicionales te permiten renombrarlos más adelante sin romper el código de nadie: como no se permite usarlos como palabra clave, eres libre de cambiarlos.

Segundo, los parámetros solo por palabra clave obligan a quien llama a ser explícito con lo que pasa, lo que hace las llamadas más legibles cuando tienes varias opciones o ajustes opcionales.

En resumen, es una forma limpia de reforzar la intención.

Funciones anónimas en Python

Las funciones anónimas también se llaman funciones lambda en Python porque, en lugar de declararlas con la palabra clave estándar def, usas la palabra lambda.

En el bloque de DataCamp Light anterior, lambda x: x*2 es la función anónima o lambda. x es el argumento y x*2 es la expresión o instrucción que se evalúa y devuelve. Lo especial de esta función es que no tiene nombre, a diferencia de los ejemplos que has visto en la primera parte de este tutorial. Si tuvieras que escribir la función anterior como UDF, el resultado sería:

def double(x):
  return x*2

Veamos otro ejemplo de función lambda con dos argumentos:

Usas funciones anónimas cuando necesitas una función sin nombre durante un corto periodo y que se crea en tiempo de ejecución. Contextos típicos: cuando trabajas con filter(), map() y reduce():

La función filter() filtra, como su nombre indica, la lista de entrada original my_list según el criterio >10. Con map(), en cambio, aplicas una función a todos los elementos de my_list. En este caso, multiplicas todos por 2.

Ten en cuenta que la función reduce() forma parte de la librería functools. La aplicas de forma acumulativa a los elementos de my_list, de izquierda a derecha, y reduces la secuencia a un único valor, 55 en este caso.

Usar main() como función en Python

Si tienes experiencia con otros lenguajes como Java, sabrás que la función main es necesaria para ejecutar funciones. Como has visto arriba, en Python no es estrictamente necesario. Aun así, incluir una función main() en tu programa puede ayudarte a estructurar el código de forma lógica: todos los componentes más importantes quedan contenidos dentro de esa función.

Puedes definir fácilmente una función main() y llamarla como has hecho con el resto de funciones anteriores:

Sin embargo, tal y como está ahora, el código de tu main() se ejecutará cuando lo importes como módulo. Para evitarlo, llama a main() solo cuando __name__ == '__main__'.

Eso significa que el bloque anterior queda así:

Nota: además de __main__, también existe la función __init__, que inicializa una instancia de una clase u objeto. En pocas palabras, actúa como constructor o inicializador y se llama automáticamente cuando creas una nueva instancia. Con esa función, el nuevo objeto se asigna al parámetro self, que ya viste en este tutorial. Observa este ejemplo:

class Dog:
    """A simple Dog class.

    Args:
        legs: Number of legs so that the dog can walk.
        color: The color of the fur.
    """

    def __init__(self, legs: int, color: str) -> None:
        self.legs = legs
        self.color = color

    def bark(self) -> str:
        return "bark" * 2

if __name__ == "__main__":
    dog = Dog(4, "brown")
    print(dog.bark())