Subscript access to python numpy array

Subscript access to numpy array

Extraction of array 1 element

>>> import numpy as np
>>> arr = np.arange(0,11)
>>> arr
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
>>> arr[5]
5

Extraction of array 1 element

>>> import numpy as np
>>> arr = np.arange(0,11)
>>> arr
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
>>> arr[5]
5

Slice the array

>>> arr[0:5]
array([0, 1, 2, 3, 4]) #5 not included

Substitute by specifying slice

>>> arr[0:5]=100
array([100, 100, 100, 100, 100,   5,   6,   7,   8,   9,  10])

I intend to make an array copy, but refer to it (even if I don't use slices)

>>> slice_of_arr = arr[0:5]
>>> slice_of_arr
array([100, 100, 100, 100, 100])
>>> slice_of_arr[:]=10
>>> slice_of_arr
array([10, 10, 10, 10, 10])
>>> arr
array([10, 10, 10, 10, 10,  5,  6,  7,  8,  9, 10]) #Change to original array

Array copy

>>> copy_arr = np.copy(arr[0:5])
>>> copy_arr
array([10, 10, 10, 10, 10])
>>> copy_arr[:] = 0
>>> arr
array([ 0, 10, 10, 10, 10,  5,  6,  7,  8,  9, 10]) #No effect on the original array

Two-dimensional subscript

>>> arr_2d = np.array(([0,1,2],[10,11,12],[20,21,22]))
>>> arr_2d 
array([[ 0,  1,  2],
       [10, 11, 12],
       [20, 21, 22]])
>>> arr_2d[0] #Take out the 0th line
array([0, 1, 2])

#Individual access arr_2d[row][col] or arr_2d[row,col]
>>> arr_2d[1][0] 
10
>>> arr_2d[1,0]
10 

Subscript utilization application example

>>> arr2d = np.zeros((10,10))
>>> arr2d
array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.]])
>>> arr_length = arr2d.shape[1] #Array length
>>> for i in range(arr_length): arr2d[i] = i
>>> arr2d
array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.],
       [ 1.,  1.,  1.,  1.,  1.,  1.,  1.,  1.,  1.,  1.],
       [ 2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.],
       [ 3.,  3.,  3.,  3.,  3.,  3.,  3.,  3.,  3.,  3.],
       [ 4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.],
       [ 5.,  5.,  5.,  5.,  5.,  5.,  5.,  5.,  5.,  5.],
       [ 6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.],
       [ 7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.],
       [ 8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.],
       [ 9.,  9.,  9.,  9.,  9.,  9.,  9.,  9.,  9.,  9.]])

>>> arr2d[[2,4,6,8]] #Extract only even rows
array([[ 2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.],
       [ 4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.],
       [ 6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.],
       [ 8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.,  8.]])
>>> arr2d[[6,4,2,7]] #Take out in a different order
array([[ 6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.,  6.],
       [ 4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.,  4.],
       [ 2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.,  2.],
       [ 7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.,  7.]])