import numpy as np
import numpy.linalg as npla
import scipy.linalg as spla
import matplotlib.pyplot as pt
from time import time
n_values = (10**np.linspace(1, 3.25, 15)).astype(np.int32)
n_values
for name, f in [
("lu", spla.lu_factor),
("qr", npla.qr),
("svd", npla.svd)
]:
times = []
print("----->", name)
for n in n_values:
print(n)
A = np.random.randn(n, n)
start_time = time()
f(A)
times.append(time() - start_time)
pt.plot(n_values, times, label=name)
pt.grid()
pt.legend(loc="best")
pt.xlabel("Matrix size $n$")
pt.ylabel("Wall time [s]")