数组中元素替换的问题

xiannvjiejie

请教如题的问题，为了方便描述，见图
  F: N, x) i- U! r8 l
( y# _+ ?$ e3 x; f# A) `有一个由0和1随机位置组成的A矩阵，现在想将A矩阵中的1和0依次替换成B矩阵和C矩阵中的元素，替换方式为图示箭头所示：B矩阵中的元素依次替换A矩阵中每个1的位置，同理，C矩阵中的元素依次替换A矩阵中每个0的位置，得到新D矩阵。

不知道是否有什么函数可以快速实现，我用for语句没有写出来。
2 b' i9 z  @' C5 y" b
8 H1 C) x% O& c) T, @0 n3 p非常感谢。

modengxian111

使用逻辑索引赋值就可以达到这个要求。示例代码如下：
8 q: F) [# Z* m+ P7 J>> A=[0,1,0,0,1,1,0,0,1]
A =
& d2 T% [  t& q& \7 H& p     0     1     0     0     1     1     0     0     1
: s  y: m% f) S- Q$ z# z* ~: c>> B=[2,1,4,3]
B =
% Z- R- [$ |( i9 c8 @! X     2     1     4     3
>> C=[7,5,9,6,8]
C =
1 x# z: ]' o3 u& c     7     5     9     6     8
, h* J# c' ~  j5 Q$ l>> A(A==1)=B
A =
  a" p. R" J  A( k/ A     0     2     0     0     1     4     0     0     3
' o! X+ s+ A- o" W>> A(A==0)=C
* g/ u" b4 L0 G- [' z' ^1 [; `A =
( Q- H( m% g# ^6 d& h( {     7     2     5     9     1     4     6     8     3