我对矩阵分配有疑问。
说我有三个矩阵A,B和C,我想根据规则将矩阵C的元素分配给A和B的元素
C[i,j] = A[i,j] if abs(C[i,j] - A[i,j]) < abs(C[i,j] - B[i,j])
C[i,j] = B[i,j] if abs(C[i,j] - A[i,j]) > abs(C[i,j] - B[i,j])
C[i,j] = 0 if abs(C[i,j] - A[i,j]) == abs(C[i,j] - B[i,j])
如何在没有for循环的情况下编写它?
非常感谢你的帮助。
答案 0 :(得分:5)
我认为Dan Becker有正确的想法,但重新计算abs(C-B)
和abs(C-A)
意味着比较更新的矩阵,而不是原始 1。
我不认为这是你想要的,所以这是他的方法的更正版本:
CmA = abs(C-A);
CmB = abs(C-B);
ind = Cma < CmB; C(ind) = A(ind);
ind = CmA > CmB; C(ind) = B(ind);
C(CmA == CmB) = 0;
答案 1 :(得分:1)
我认为您需要以下内容:
ind = abs(C - A) < abs(C - B) ; C(ind) = A(ind);
ind = abs(C - A) > abs(C - B) ; C(ind) = B(ind);
ind = abs(C - A) == abs(C - B) ; C(ind) = 0;