有没有办法让我修改这段代码，让它产生一个可以存储为2d列表的输出？

output = []

# 假设原始输出为一个一维列表
original_output = [1, 2, 3, 4, 5, 6, 7, 8, 9]

# 定义每行的元素个数
row_length = 3

# 将原始输出按照每行元素个数进行切分，并存储为2D列表
for i in range(0, len(original_output), row_length):
    row = original_output[i:i+row_length]
    output.append(row)

# 打印输出
for row in output:
    print(row)

[1, 2, 3]
[4, 5, 6]
[7, 8, 9]