数组中元素替换的问题

xiannvjiejie

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

+ @% W# |9 f- ^有一个由0和1随机位置组成的A矩阵，现在想将A矩阵中的1和0依次替换成B矩阵和C矩阵中的元素，替换方式为图示箭头所示：B矩阵中的元素依次替换A矩阵中每个1的位置，同理，C矩阵中的元素依次替换A矩阵中每个0的位置，得到新D矩阵。
" {* @. o6 x9 Q1 F* j- Q- N" l
; p. [5 _  S0 R: Z# n不知道是否有什么函数可以快速实现，我用for语句没有写出来。
" K; i0 N9 g1 c3 Q% Q9 l, C( }
6 ^9 p; j: m% Y1 M4 s非常感谢。
# z1 Z2 l( K8 X' x. i( k/ P- n
; A; P4 u0 J# _) @" p9 `* u4 B

modengxian111

4 p  P, @1 f& }- s使用逻辑索引赋值就可以达到这个要求。示例代码如下：
0 m* B- Y1 Y$ f' @! K2 r* ]>> A=[0,1,0,0,1,1,0,0,1]
! n0 i9 m9 o& M1 w6 R2 s; bA =
" k; w& b) ^2 \& A  ?     0     1     0     0     1     1     0     0     1
8 c4 c& H9 m; n, t# F: [) f>> B=[2,1,4,3]
B =
     2     1     4     3
7 w6 M: F" U) C* V>> C=[7,5,9,6,8]
C =
. U! O) K) y2 l% }) `1 N& L% A9 ~     7     5     9     6     8
>> A(A==1)=B
A =
* B& ?( _$ U( p, K# x. ]. @! j     0     2     0     0     1     4     0     0     3
5 M( L$ s* K+ C$ l>> A(A==0)=C
$ o" |. a6 h: a9 ]" _8 GA =
     7     2     5     9     1     4     6     8     3