如何查询给定范围内矩阵元素的计数

def count_matrix_elements(matrix, row1, col1, row2, col2):
    count = 0
    for i in range(row1, row2+1):
        for j in range(col1, col2+1):
            if i >= 0 and i < len(matrix) and j >= 0 and j < len(matrix[0]):
                count += 1
    return count

# 示例矩阵
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 查询范围为(0, 0)到(1, 1)
count = count_matrix_elements(matrix, 0, 0, 1, 1)
print("给定范围内矩阵元素的计数为:", count)