如何调整Numpy数组的大小以使用由数组每行中的值确定的组合来添加/替换行

import numpy as np

# 原始数组
original_array = np.array([[1, 2, 3],
                           [4, 5, 6],
                           [7, 8, 9]])

# 调整后的行数
new_rows = original_array[:, 0] * original_array[:, 1]

# 调整后的列数
new_cols = original_array[:, 2]

# 创建新数组
new_array = np.empty((new_rows.shape[0], new_cols.shape[0]))

# 复制每一行
for i in range(original_array.shape[0]):
    row = original_array[i]
    repeated_row = np.tile(row, (new_rows[i], 1))
    new_array = np.vstack((new_array, repeated_row))

# 组合新数组和原始数组
result_array = np.hstack((original_array, new_array))

print(result_array)

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