Extract Specific RANGE of columns in numpy array Python

前端 未结 2 1347
长情又很酷
长情又很酷 2020-12-30 04:14

I have an array :

e = np.array([[ 0,  1,  2,  3, 5, 6, 7, 8],
              [ 4,  5,  6,  7, 5, 3, 2, 5],
              [ 8,  9, 10, 11, 4, 5, 3, 5]])
         


        
2条回答
  •  天涯浪人
    2020-12-30 04:42

    Numpy row and column indices start counting at 0.

    The rows are specified first and then the column with a comma to separate the row from column.

    The ":" (colon) is used to shortcut all rows or all columns when it is used alone.

    When the row or column specifier has a range, then the ":" is paired with numbers that specify the inclusive start range and the exclusive end range.

    For example

    import numpy as np
    np_array = np.array( [ [ 1, 2, 3, ],
                           [ 4, 5, 6, ],
                           [ 7, 8, 9  ] ]  )
    
    first_row = np_array[0,:]
    first_row
    output: array([1, 2, 3])
    
    last_column = np_array[:,2]
    last_column
    output: array([3, 6, 9])
    
    first_two_vals = np_array[0,0:2]
    first_two_vals
    output: array([1, 2])
    

提交回复
热议问题