当除数为零时保留旧值

Question

我有一个长度为N的向量D和一个形状为N*M的矩阵a。向量D有一些零元素。我在做这个手术：

D = D.reshape(-1,1)
A / D

因为D有一些零，所以我将得到一个被零除的误差。我想要的不是修改A的值，在这里被零除。如果A[i，j]/0，那么我希望A[i，j]保持它的值。怎么做？你知道吗

Answer 1

试试这个：

A = np.where(D != 0, A / D, A)