Slicing arrays in Numpy / Scipy

前端 未结 3 795
栀梦
栀梦 2020-12-03 13:54

I have an array like:

a = array([[1,2,3],[3,4,5],[4,5,6]])

What\'s the most efficient way to slice out a 1x2 array out of this that has onl

3条回答
  •  一整个雨季
    2020-12-03 14:27

    Two dimensional numpy arrays are indexed using a[i,j] (not a[i][j]), but you can use the same slicing notation with numpy arrays and matrices as you can with ordinary matrices in python (just put them in a single []):

    >>> from numpy import array
    >>> a = array([[1,2,3],[3,4,5],[4,5,6]])
    >>> a[:,1:]
    array([[2, 3],
           [4, 5],
           [5, 6]])
    

提交回复
热议问题