Question

假设我有这个矩阵：

mx1 = np.matrix([[2,0,9],[0,5,8],[1,2,0]])

[[2 0 9]
 [0 5 8]
 [1 2 0]]

我想只用来自另一个矩阵（相同尺寸）的值替换0值：

mx2 = np.matrix([[5,6,2],[3,3,7],[0,1,6]])

[[5 6 2]
 [3 3 7]
 [0 1 6]]

结果矩阵应为：

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

请不要循环。想法？

Answer 1

使用np.where的单线：

np.where(mx1 == 0, mx2, m1)

为这些文档添加书签，它会为您节省下来。