根据一个矩阵中的值，提取另外一个矩阵对应位置的值

Question

假设我有两个矩阵a1,a2

a1 = RandomReal[10, {3, 3}];
MatrixForm[a1]
a2 = RandomInteger[1, {3, 3}];
a2 // MatrixForm

我想在提取a1中的值，只有相同位置上a2的值为1，才提取对应位置上a1的值，请问该怎么做？

Answer 1

先把a1中不为1的置0，再对应数乘吧。得到的矩阵就是a1经a2过滤后的矩阵了。

a3 = Map[Replace[_?(# != 1 &) -> 0], a2, Infinity]
a1*a3;
% // MatrixForm

只知道思路，代码不太会，大致是这样。

Comments

——请问如何更改自己的用户名、头像等信息呢？我是QQ登录的。

Answer 2

你是不是找：Pick

a1 = RandomReal[10, {3, 3}];
a2 = RandomInteger[1, {3, 3}];
Pick[a1, a2, 1]