标签： python numpy
我有一个长度为N的向量D和一个形状为N * M的矩阵A.矢量D有一些零元素。我正在做这个操作：
D = D.reshape(-1,1) A / D
由于D有一些零，所以我会得到零除错误。我想要的是不要修改除以零的A值。所以，如果A [i，j] / 0，那么我希望A [i，j]保持其值。怎么做？
答案 0 :(得分：1)
试试这个：
A = np.where(D != 0, A / D, A)