Numpy: Sắp xếp lại mảng dựa trên mảng chỉ mục

Question

Tôi đang tìm một giải pháp một dòng có thể giúp tôi thực hiện những việc sau.

Giả sử tôi có

array = np.array([10, 20, 30, 40, 50]) 

Tôi muốn sắp xếp lại nó dựa trên một trật tự đầu vào. Nếu có một chức năng NumPy gọi arrange, nó sẽ làm như sau:

newarray = np.arrange(array, [1, 0, 3, 4, 2]) 
print newarray 

    [20, 10, 40, 50, 30] 

Chính thức, nếu mảng được sắp xếp lại là mxn, và "chỉ số" mảng là 1 xn, thứ tự sẽ được xác định bởi mảng được gọi là "chỉ mục".

Numpy có chức năng như thế này không?

Answer 1

Bạn chỉ có thể sử dụng "chỉ số" danh sách của bạn trực tiếp, như, tốt, một mảng chỉ mục:

>>> arr = np.array([10, 20, 30, 40, 50]) 
>>> idx = [1, 0, 3, 4, 2] 
>>> arr[idx] 
array([20, 10, 40, 50, 30]) 

Nó có xu hướng để được nhanh hơn nhiều nếu idx đã là một ndarray và không phải là một list, mặc dù nó sẽ hoạt động theo một trong hai cách:

>>> %timeit arr[idx] 
100000 loops, best of 3: 2.11 µs per loop 
>>> ai = np.array(idx) 
>>> %timeit arr[ai] 
1000000 loops, best of 3: 296 ns per loop