如何在没有numpy和用户输入的Python上创建一个三对角矩阵？

import random
import sys

# 获取用户输入的矩阵大小
n = int(input("请输入矩阵的大小："))

# 创建一个空的二维列表作为矩阵容器
matrix = [[0] * n for _ in range(n)]

# 使用循环遍历矩阵的行和列，并根据三对角矩阵的特点赋值矩阵元素
for i in range(n):
    for j in range(n):
        if i == j:
            matrix[i][j] = random.randint(1, 9)  # 主对角线元素
        elif abs(i - j) == 1:
            matrix[i][j] = random.randint(1, 9)  # 次对角线元素

# 打印输出生成的三对角矩阵
for row in matrix:
    print(row)