我有一个矩阵,例如
A = [ 1 2 3; 4 5 6; 7 8 9] ;
Run Code Online (Sandbox Code Playgroud)
和一个大小为1x3的向量,它指定每行中哪个元素是我正在寻找的元素 - 即If
vector = [ 1 2 1 ]
Run Code Online (Sandbox Code Playgroud)
那么期望的输出是
[ 1 5 7 ]
Run Code Online (Sandbox Code Playgroud)
因为1是第1行中的第1个元素,5是第2行中的第2个7元素,并且是第3行中的第1个元素.
我该如何实现这一目标?无法找到内置功能来做到这一点,这让我感到惊讶.