Nhận cột khác nhau trong mỗi hàng

Question

Tôi muốn lấy một cột khác nhau trong mỗi hàng từ một ma trận. Ví dụ:

A = [1,2;1,4;5,2] 
B = [2;2;1] 

đầu ra nên năng suất:

out = [2;4;5] 

Vì vậy, trong ngắn hạn: A là ma trận và B có các chỉ số cho colums mỗi hàng. Làm thế nào tôi có thể làm điều này mà không cần sử dụng một vòng lặp (nếu nó có thể)?

Answer 1

Bạn có thể sử dụng sub2ind để chuyển đổi các chỉ số (i, j) để chỉ số tuyến tính trong ma trận A

idx = sub2ind(size(A), 1:size(A, 1), B'); 
A(idx) 

ans = 

2  4  5 

mà làm việc giả định rằng vector B có nhiều entry như có một dòng ở A. Trong sub2ind trên tạo chỉ mục tuyến tính cho mỗi cặp số hàng tiếp theo (1: kích thước (A, 1)) và số cột được cung cấp trong B.

Answer 2

Bạn có thể thực hiện theo số diag(A(:,B)), tuy nhiên, chỉ có một vòng lặp. Tôi không biết làm thế nào để làm điều đó mà không có bất kỳ vòng lặp.