Example code for interview with Robin

python_example_code.py

import numpy as np
from scipy.signal import convolve2d


def func(arr):
    """
    Arguments:
        arr: a numpy array (any numeric dtype, any dimensions)

    Return value:
        a single floating point number
    """
    counts, bins = np.histogram(arr)

    i = np.argmax(counts)

    l = bins[i]
    u = bins[i + 1]

    return (l + u) / 2.0


def f(arr):
    """
    Arguments:
        arr: 2D numpy array, with dtype=bool

    Return value:
        numpy array of the same size and dtype as `arr`
    """
    kernel = np.array([[1, 1, 1],
                       [1, 0, 1],
                       [1, 1, 1]])

    result = np.bitwise_and(convolve2d(arr, kernel, mode='same') == 1,
                            arr)

    return result