使用基于第三个矩阵 C 的条件语句从矩阵 A 到矩阵 B 的矩阵分配

Question

我有两个问题，如果你可以回答：

Q1) 我有一个矩阵choice，其中每个人做出 4 个可能的选择，分别表示为 1、2、3 和 4。 我有三个矩阵A1、A2、A3，其中包含每个人和每个时间段的收入信息。假设我有 n 个人和 t 个时间段，所以 A1、A2、A3 是 n-by-t 和 choice 是 n-by-t。

现在我想创建另一个矩阵B，其中B 将根据选择矩阵中的值从A 中选择元素，即如果choice(n,t)==1，则B(n,t) = A1(n,t)。如果是choice(n,t)==2，那么B(n,t) = A2(n,t)，以此类推。

我已经尝试了 for 循环和 if 语句，但我无法做到。请帮忙。

Q2) 我有一个收入矩阵A。 A 是 n×t 维度。有些人收入低，有些人收入高。假设任何收入

Answer 1

第一季度：

C = choice %else the code gets too long
B = A1 .* (C==1) + A2 .* (C==2) + A3 .* (C==3)

如果您只有 A1 A2 A3，我不确定您想如何处理 B 中的值“4”，但这应该可以。

[编辑]： 如果选项是“4”，那么对于上面定义的 B i，B 的那个元素将为 0。

第二季度： 这个有点模糊。也许这就是你想要的：

HighIncome = A > 1000
LowIncome = A <= 1000

如果不这样做，请更准确地解释您的目标。

[编辑]： 根据您对 Q2 稍微不那么含糊的解释，听起来您不希望这样：

A_high_income = A .* (A > 1000)
A_low_income = A .* (A <= 1000)
CHOICE_high_income = choice .* (A > 1000)
CHOICE_high_income = choice .* (A <= 1000)

高收入矩阵在低收入位置处为零，反之亦然。 恕我直言，这并没有多大意义，但这是我能得到的最接近你的描述。 如果不这样做，请按照我下面评论中的说明并发布一些示例。

Answer 2

Q1：你可以使用三个简单的语句和一些逻辑索引。

B = A1;
B(choice == 2) = A2(choice == 2);
B(choice == 3) = A3(choice == 3);

Q2：要根据收入将 A 和选择分成两部分，您首先找到“低收入”行的索引并使用它从矩阵中获取行。

lowIncomeNdx = any(A < 1000, 2);
lowIncome = A(lowIncomeNdx, :);
lowIncomeChoice = choice(lowIncomeNdx, :);
highIncome = A(~lowIncomeNdx, :);
highIncomeChoice = choice(~lowIncomeNdx, :);