数组中元素替换的问题

xiannvjiejie

请教如题的问题，为了方便描述，见图
8 ^. T" P; T) I/ D1 I, _- T
" g6 N# |" D9 }& ~有一个由0和1随机位置组成的A矩阵，现在想将A矩阵中的1和0依次替换成B矩阵和C矩阵中的元素，替换方式为图示箭头所示：B矩阵中的元素依次替换A矩阵中每个1的位置，同理，C矩阵中的元素依次替换A矩阵中每个0的位置，得到新D矩阵。

7 T  H) Z, G2 X4 ]( g9 }2 ?不知道是否有什么函数可以快速实现，我用for语句没有写出来。
2 D- {8 `' {2 w$ s3 N2 |; b
- `+ W, s7 p" K" O非常感谢。

modengxian111

使用逻辑索引赋值就可以达到这个要求。示例代码如下：
>> A=[0,1,0,0,1,1,0,0,1]
* T. H0 d. i+ N% f- v5 @A =
     0     1     0     0     1     1     0     0     1
' L) ]0 g. E: B7 `: Y. A>> B=[2,1,4,3]
; F* d# b! \1 n9 Y! eB =
! ~4 U( F( J7 E' V' ]  @     2     1     4     3
5 ?3 g7 N) j7 p6 p>> C=[7,5,9,6,8]
C =
) A. M2 h! ]8 s& G; x8 E; {% y: l# k     7     5     9     6     8
6 W- e- V/ ]$ G7 }, N>> A(A==1)=B
' J, C& n- a. v7 S" h/ {A =
  @4 y- a6 b5 d. w# O     0     2     0     0     1     4     0     0     3
>> A(A==0)=C
1 F- L- L& R+ w+ HA =
4 T; U% x, n: D% c     7     2     5     9     1     4     6     8     3