Functools

Partial functions

from functools import partial
import numpy as np
import matplotlib.pyplot as plt

def general_quadratic_function(x: np.array, params: tuple):
    a,b,c = params
    return a*(x**2) + (b*x) + c

our_quadratic_function = partial(general_quadratic_function, params=(1,2,3))

our_quadratic_function(np.arange(10))

array([  3,   6,  11,  18,  27,  38,  51,  66,  83, 102])

Plot functions

np.arange(0,10,0.1)

array([0. , 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1. , 1.1, 1.2,
       1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2. , 2.1, 2.2, 2.3, 2.4, 2.5,
       2.6, 2.7, 2.8, 2.9, 3. , 3.1, 3.2, 3.3, 3.4, 3.5, 3.6, 3.7, 3.8,
       3.9, 4. , 4.1, 4.2, 4.3, 4.4, 4.5, 4.6, 4.7, 4.8, 4.9, 5. , 5.1,
       5.2, 5.3, 5.4, 5.5, 5.6, 5.7, 5.8, 5.9, 6. , 6.1, 6.2, 6.3, 6.4,
       6.5, 6.6, 6.7, 6.8, 6.9, 7. , 7.1, 7.2, 7.3, 7.4, 7.5, 7.6, 7.7,
       7.8, 7.9, 8. , 8.1, 8.2, 8.3, 8.4, 8.5, 8.6, 8.7, 8.8, 8.9, 9. ,
       9.1, 9.2, 9.3, 9.4, 9.5, 9.6, 9.7, 9.8, 9.9])

def plot_function(func, x=np.arange(0,10,0.1), ax=None, **kwargs):
    if 'figsize' not in kwargs:
        kwargs['figsize'] = (3,3)
    if ax is None:
        ax = plt.subplots(**kwargs)[1]

    y = func(x)
    ax.plot(x, y)

plot_function(our_quadratic_function)

Typing

from typing import List, TypeVar
MyType = TypeVar('MyType', str, int )

def foo(a: List[int]):
    print(a)