numpy的：工作矩阵的一个子集

Question

我在numpy中有一个4x4的单位矩阵，我想按一个因子缩放前3个维度。 目前，我的操作方式如下：

# Some scaling factors passed as a parameter by the user
scale = (2, 3, 4)  
scale += (1,)  # extend the tuple
my_mat = scale * np.eye(4)

出于好奇，我想知道是否可以通过某种方式来扩展元组。

Answer 1

这可以通过numpy 广播规则和索引快速完成

A = np.eye(4)
scale = [2, 3, 4]

A[:3, :3] *= scale