假设有一个矩阵matrix和两个列表rows和cols,需要按照rows和cols中的元素扩展矩阵matrix。
代码示例:
import numpy as np
# 定义原矩阵
matrix = np.array([[1, 2], [3, 4]])
# 定义扩展列表
rows = [0, 2]
cols = [1, 3]
# 计算扩展后的矩阵大小
new_rows = max(rows) + 1
new_cols = max(cols) + 1
# 创建扩展后的矩阵并赋初值
new_matrix = np.zeros((new_rows, new_cols), dtype=matrix.dtype)
new_matrix[:matrix.shape[0], :matrix.shape[1]] = matrix
# 输出扩展后的矩阵
print(new_matrix)
输出结果为:
array([[1, 2, 0, 0],
[3, 4, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0]])
解释: 代码首先将原矩阵和扩展列表给出,然后计算出扩展后的矩阵大小。之后,新建一个大小为扩展后矩阵大小的零矩阵,并将原矩阵的值复制到新矩阵的对应位置。最后输出扩展后的矩阵。