derivative_of_x.py

def derivative(func, x):
    h = 1e-10
    x_d = np.full_like(x, 0) # the derivatives
    x_c = x.copy()
    for idx, value in np.ndenumerate(x):
        x_c[idx] += h
        x_d[idx] = (func(x_c) - func(x)) / h
        x_c[idx] -= h
    return x_d