import numpy as np
import scipy.sparse
import scipy.sparse.linalg
from scipy.sparse import csr_matrix
from scipy.sparse import spdiags


def matrixA(n):
    data = np.array([-np.ones(n),2*np.ones(n),-np.ones(n)])
    diags = np.array([-1,0, 1])
    A=spdiags(data, diags, n, n);
    return A

n=10

A=matrixA(n)
b=np.ones(n)
print(A@b)