Làm thế nào để có được các giá trị từ một mảng NumPy sử dụng nhiều chỉ số

Question

Tôi có một mảng NumPy trông như thế này:

arr = [100.10, 200.42, 4.14, 89.00, 34.55, 1.12] 

Làm thế nào tôi có thể nhận được nhiều giá trị từ mảng này bởi chỉ số:

Ví dụ , làm thế nào tôi có thể nhận được các giá trị tại các vị trí chỉ mục 1, 4 và 5?

Tôi đã cố gắng một cái gì đó như thế này, đó là không chính xác:

arr[1, 4, 5] 

Answer 1

Hãy thử như thế này:

>>> arr = np.array([100.10, 200.42, 4.14, 89.00, 34.55, 1.12]) 
>>> arr[[1,4,5]] 
array([ 200.42, 34.55, 1.12]) 

Và đối với mảng đa chiều:

>>> arr = np.arange(9).reshape(3,3) 
>>> arr 
array([[0, 1, 2], 
     [3, 4, 5], 
     [6, 7, 8]]) 
>>> arr[[0, 1, 1], [1, 0, 2]] 
array([1, 3, 5]) 

Answer 2

bạn đã gần

>>> print arr[[1,4,5]] 
[ 200.42 34.55 1.12] 

Answer 3

Một giải pháp khác là sử dụng np.take theo quy định tại https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.take.html

a = [4, 3, 5, 7, 6, 8] 
indices = [0, 1, 4] 
np.take(a, indices) 
# array([4, 3, 6])