给我一段解三对角方程组的追赶法的python代码

以下是解三对角方程组的追赶法的Python代码：

import numpy as np

def tridiagonal_solver(a, b, c, d):
    """
    Solves a tridiagonal system of equations using the Thomas algorithm.
    :param a: array of lower diagonal elements
    :param b: array of diagonal elements
    :param c: array of upper diagonal elements
    :param d: array of right-hand side values
    :return: array of solution values
    """
    n = len(b)
    
    # Modify the diagonal and upper diagonal arrays
    for i in range(1, n):
        m = a[i-1] / b[i-1]
        b[i] -= m * c[i-1]
        d[i] -= m * d[i-1]
    
    # Back substitution
    x = np.zeros(n)
    x[-1] = d[-1] / b[-1]
    for i in range(n-2, -1, -1):
        x[i] = (d[i] - c[i] * x[i+1]) / b[i]
    
    return x

其中，a、b、c和d分别表示三对角矩阵的下对角线、对角线、上对角线和右端向量。函数返回解向量x。