Uzyskaj inną kolumnę w każdym wierszu
Chciałbym uzyskać inną kolumnę z każdego wiersza z macierzy. Na przykład:
A = [1,2;1,4;5,2]
B = [2;2;1]
wyjście powinno dawać:
out = [2;4;5]
Krótko mówiąc:A
jest macierzą iB
ma indeksy dla kolumn w wierszu. Jak mogę to zrobić bez użycia pętli (jeśli to możliwe)?