在NumPy中,有幾種常見的數組索引方式:
索引單個元素:可以使用方括號 []
來索引單個元素,指定元素的位置。例如,對于一個二維數組 arr
,可以使用 arr[row_index, column_index]
來獲取指定位置的元素。
切片索引:使用切片操作 :
可以獲取數組的一個子集。切片操作可以用于一維和多維數組,以及不同軸向上的切片。例如,對于一個一維數組 arr
,可以使用 arr[start:end:step]
來獲取從 start
到 end-1
的元素,步長為 step
。對于一個二維數組 arr
,可以使用 arr[start1:end1:step1, start2:end2:step2]
來獲取行范圍為 start1:end1-1
,列范圍為 start2:end2-1
的子集。
整數數組索引:可以使用一個整數數組來索引數組的元素。整數數組的每個元素表示要獲取的元素在對應軸上的位置。例如,對于一個一維數組 arr
,可以使用一個整數數組 idx
來獲取 arr[idx]
中的元素。對于一個二維數組 arr
,可以使用兩個整數數組 idx1
和 idx2
來獲取 arr[idx1, idx2]
中的元素。
布爾數組索引:可以使用一個布爾數組來索引數組的元素。布爾數組的每個元素表示對應位置上的元素是否被選中。例如,對于一個一維數組 arr
,可以使用一個布爾數組 mask
來獲取 arr[mask]
中為 True
的元素。對于一個二維數組 arr
,可以使用一個布爾數組 mask
來獲取 arr[mask]
中為 True
的元素。
這些索引方式可以組合使用,以滿足不同的需求。