数组中元素替换的问题

xiannvjiejie

请教如题的问题，为了方便描述，见图

有一个由0和1随机位置组成的A矩阵，现在想将A矩阵中的1和0依次替换成B矩阵和C矩阵中的元素，替换方式为图示箭头所示：B矩阵中的元素依次替换A矩阵中每个1的位置，同理，C矩阵中的元素依次替换A矩阵中每个0的位置，得到新D矩阵。

; U, R2 h. J6 k. F4 d不知道是否有什么函数可以快速实现，我用for语句没有写出来。

非常感谢。
1 \2 d8 S1 A6 |. L8 A4 a
2 A. `7 H" X! e8 e) P' B7 m

modengxian111

2 L+ l6 X( d" ]3 X0 m9 u使用逻辑索引赋值就可以达到这个要求。示例代码如下：
. ?' O9 s+ ~+ ~# A) l>> A=[0,1,0,0,1,1,0,0,1]
8 I. s+ i) X' Q' I; l& W. ^5 X3 |A =
/ Y% C: ]$ [* r$ [8 P     0     1     0     0     1     1     0     0     1
>> B=[2,1,4,3]
  V- m  S' J( RB =
     2     1     4     3
, d2 V7 H7 P" r) v3 L! P: ?* Y>> C=[7,5,9,6,8]
2 D2 o  M+ ~( N: g  y- h1 pC =
     7     5     9     6     8
>> A(A==1)=B
A =
     0     2     0     0     1     4     0     0     3
. q$ i( m8 r5 P8 q1 `>> A(A==0)=C
. c7 Y5 q6 G( \" @. F+ S; Y6 j0 P" qA =
7 F0 Q$ X. L& \! a7 M' Z, {     7     2     5     9     1     4     6     8     3