numpy的基本API（三）——索引

numpy的基本索引API  

　　iwehdio的博客园：https://www.cnblogs.com/iwehdio/

 

1、单个元素的索引

　　对于一维数组，索引方式与内置的List相同。正索引从0开始，负索引从-1开始。

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

　　

　　在多维数组中也是同理。x[0,2] 与 x[0][2]是等价的，但是后者的效率更低。

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

　　

　　如果想要索引时对x的个轴不作限制。可以用冒号“：”或省略号“...”表示选取该行或列的所有数据。在这里x[0]等价于x[0, :]。

>>> x[0]
array([0, 1, 2, 3, 4])
>>> x[:, 1]
array([1, 6])

　　

　　切片方式与List类似。a:b:c表示以a为起点，b为终点，步长为c。对于多维数组同样类似，:表示选取该个维度的所有数据。由于-1表示负步长，可以使用x[::-1]来反转数组。

>>> x[:,1:3]
array([[1, 2],
       [6, 7]])
>>> x[:,::-1]
array([[4, 3, 2, 1, 0],
       [9, 8, 7, 6, 5]])

 　　

　　使用np.newaxis，可以在数组中插入某个长度为1的轴。

>>> x.shape
(3, 2)
>>> x[:,np.newaxis,:].shape
(3, 1, 2)

 

2、数组索引

 

　　可以通过整数数组定义的索引值，来自定义索引。即索引了数组x中的元素x[0,0]、x[1,1]、x[2,0]，并返回一个数组。

 

>>> x = np.array([[1, 2], [3, 4], [5, 6]])
>>> x
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> x[[0, 1, 2], [0, 1, 0]]
array([1, 4, 5])

　　

　　也可以用广播来创建索引数据。比如用newaxis或者ix_。最终都索引到了x[0,0]、x[0,2]、x[3,0]、x[3,2]。

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

  
>>> rows = np.array([0, 3])
>>> columns = np.array([0, 2])
>>> rows[:, np.newaxis]
array([[0],
          [3]])
>>> x[rows[:, np.newaxis], columns]
array([[ 0,  2],
       [ 9, 11]])       

>>> x[np.ix_(rows, columns)]
array([[ 0,  2],
       [ 9, 11]])             

 

3、布尔数组索引

 

　　使用布尔数组索引时，[ ] 内的为True或Flase或结果为True或Flase的表达式。下边第一个例子表示实际上是 [ ] 内的布尔值为True时进行索引。第二个例子表示对 x<0 为True的元素进行索引，并对其加20。第三个例子是，索引元素行的和小于等于2的行。

>>> x = np.array([1., -1., -2., 3])
>>> x[[True,False,False,True]]
array([1., 3.])

>>> x[x < 0] += 20
>>> x
array([ 1., 19., 18.,  3.])

>>> x = np.array([[0, 1], [1, 1], [2, 2]])
>>> rowsum = x.sum(-1)
>>> x[rowsum <= 2, :]
array([[0, 1],
       [1, 1]])

 

4、元组索引

 

　　用元组可以整体设置数组的索引，处理可变数量的索引。相比而言用数组索引会出现错误。以索引下面的z的z[1,1,1,1]为例。用元组只需写z[(1,1,1,1)]。用数组写z[[1,1,1,1]]，则是索引到了4个z[1]数组组成的shape为（4,1）的数组。

>>> z = np.arange(81).reshape(3,3,3,3)
>>> indices = (1,1,1,1)
>>> z[indices]
40

>>> indices = (1,1,1,slice(0,2)) 
>>> z[indices]
array([39, 40])

　　同时，也可以用元组进行切片，使用slice(a,b)函数，等价于a:b(但不能在元组中使用冒号)。

 

5、索引相关函数

　　

　　np.r_[ 'form', a, b]　　np.r_方法可以把输入的多个数组索引进行拼接。'form'是按逗号分隔的整数字符串，当只有一个整数时，表示按第几个轴进行拼接。

>>> np.r_[-1:1:6j, [0]*3, 5, 6]
array([-1. , -0.6, -0.2,  0.2,  0.6,  1. ,  0. ,  0. ,  0. ,  5. ,  6. ])

>>> np.r_['-1', a, a] 
array([[0, 1, 2, 0, 1, 2],
       [3, 4, 5, 3, 4, 5]])

>>> np.r_['0', a, a]
array([[0, 1, 2],
       [3, 4, 5],
       [0, 1, 2],
       [3, 4, 5]])

　

　　'form'整数字符串中有效的是其前三位整数，如 'q, w, e '。以下用数组 x = np.arange(24).reshape(2,3,4)为例。

　　第一个整数q表示所要拼接的是哪一个轴。若对两个相同的输入数组进行拼接，输入数组shape为(2,3,4)，则q取0,1,2所拼接的数组shape分别(4,3,4)、(2,6,4)、(2,3,8)。

>>> np.r_['0', x, x].shape
(4, 3, 4)
>>> np.r_['1', x, x].shape
(2, 6, 4)
>>> np.r_['2', x, x].shape
(2, 3, 8)

 

　　第二个整数w表示输出的最小维数，若大于输入数组的维数则补充。如果w比输入维数大，如对前例有w=4，则输出数组ans的维数为四维。默认状态下新添加的轴的位置为-1，即在输入shape之后。如对q=0，w=4有输出shape为(2,2,3,4)。如果w比输入维度大n(>1),如对前例有w=5，则输出输出数组ans的shape为(2,1,2,3,4)。

>>> np.r_['0,4', x, x].shape
(2, 2, 3, 4)
>>> np.r_['0,5', x, x].shape
(2, 1, 2, 3, 4)

 

　　第三个整数e表示新增的轴在shape中的排列方式，默认值为‘-1’表示新增轴都在原数组的shape之前，之前对第二个整数w举例时都在此种情况。e的取值范围最大为w-输入数组维度，‘0’表示新增的shape为1的轴全在输入数组的shape之后，‘1’表示新增的shape为1的轴有一个轴在输入数组的shape之前，‘2’表示新增的shape为1的轴有两个轴在输入数组的shape之前，以此类推。

>>> np.r_['0,5,0', x, x].shape
(4, 3, 4, 1, 1)
>>> np.r_['0,5,1', x, x].shape
(2, 2, 3, 4, 1)
>>> np.r_['0,5,2', x, x].shape
(2, 1, 2, 3, 4)

 

　　需要说明的是，虽然输入的‘form’整数字符串是按 q、w、e 的顺序输入的，但是numpy对参数的处理顺序却是 w、e、q。也就是说，首先按输入的w计算需要补几个shape为1的轴，然后按e对shape进行顺序设置，最后用q去索引设置更新好的shape的数组。这就是q=0,w=4,e=-1时输出shape为(2,2,3,4)的原因。

 　　如果是不同shape的数组连接，一定要注意好连接的轴的索引。

>>> y = np.arange(12).reshape(1,3,4)
>>> np.r_['1,4', x, y].shape
(1, 3, 3, 4)
>>> np.r_['2,5,2', x, y].shape
(1, 1, 3, 3, 4)

 

　　np.c_[a, b]　　np.c_方法可以把输入的索引数组返回二维数组的输出，实际上等价于np.r_['-1,2,0', a, b]。

>>> np.c_[np.array([1,2,3]), np.array([4,5,6])]
array([[1, 4],
       [2, 5],
       [3, 6]])
>>> np.r_['-1,2,0',np.array([1,2,3]), np.array([4,5,6])]
array([[1, 4],
       [2, 5],
       [3, 6]])

 

　　np.s_[ a:b:c ]　　np.s_方法可以创建索引元组，a:b:c表示从a到b步长为c的数组索引，与直接索引数组的方法类似。

>>> np.s_[2::1]
slice(2, None, 1)
>>> np.array([0, 1, 2, 3, 4])[np.s_[2::1]]
array([2, 3, 4])

>>> np.s_[:,2::1]
(slice(None, None, None), slice(2, None, 1))
>>> x = np.arange(24).reshape(2, 3, 4)
>>> x[np.s_[:,2::1]]
array([[[ 8,  9, 10, 11]],
       [[20, 21, 22, 23]]])

 

　　np.nonzero(a)　　np.nonzero方法返回数组中非零元素的索引。也可以用于查找数组中满足某条件的元素的索引。

>>> x = np.array([[3, 0, 0], [0, 4, 0], [5, 6, 0]])
>>> x
array([[3, 0, 0],
       [0, 4, 0],
       [5, 6, 0]])
>>> np.nonzero(x)
(array([0, 1, 2, 2]), array([0, 1, 0, 1]))
>>> x[np.nonzero(x)]
array([3, 4, 5, 6])

>>> a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> a > 3
array([[False, False, False],
       [ True,  True,  True],
       [ True,  True,  True]])
>>> np.nonzero(a > 3)
(array([1, 1, 1, 2, 2, 2]), array([0, 1, 2, 0, 1, 2]))
>>> a[np.nonzero(a > 3)]
array([4, 5, 6, 7, 8, 9])

 

　　np.where( condition, x, y )　　np.where方法根据condition的条件，为真时返回x中对应的值，为假时返回y中对应的值。

>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a < 5, a, 10*a)
array([ 0,  1,  2,  3,  4, 50, 60, 70, 80, 90])

>>> x, y = np.ogrid[:3, :4]
>>> np.where(x < y, x, 10 + y) 
array([[10,  0,  0,  0],
       [10, 11,  1,  1],
       [10, 11, 12,  2]])

 

　　np.ix_(a, b, c)　　np.ix_方法根据输入的一维数组a、b、c构造索引网格。

>>> x = np.arange(24).reshape(2, 3, 4)
>>> index = np.ix_([0, 1], [1, 2],[2, 3])
>>> index
(array([[[0]],
        [[1]]]), 
 array([[[1],
         [2]]]),
 array([[[2, 3]]]))
>>> x[index]
array([[[ 6,  7],
        [10, 11]],
       [[18, 19],
        [22, 23]]])

 

　　np.take(a, index, axis=None)　　np.take方法按轴索引元素。当axis=None时，将数组看作是展平的。当axis指定时，输出对该轴索引的结果。index输入的只是对指定轴的一维索引，如果输入为二维数组，会导致输出为二维数组中一维数组索引结果的数组拼接。

>>> x = np.arange(24).reshape(2, 3, 4)
>>> np.take(x,[0,5,10,15,20])
array([ 0,  5, 10, 15, 20])

>>> np.take(x, [0,1], axis=1)
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7]],
       [[12, 13, 14, 15],
        [16, 17, 18, 19]]])

>>> np.take(a, [[0, 1], [2, 3]])
array([[4, 3],
       [5, 7]])

 

 　　np.compress(index, a, axis=None)　　np.compress方法返回按给定轴的数组的选定切片。index表示在该轴上的布尔数组切片索引，a表示输入数组，axis表示选定的轴。

>>> x = np.arange(24).reshape(2,3,4)
>>> np.compress([False, False, True, True],x,axis=2)
array([[[ 2,  3],
        [ 6,  7],
        [10, 11]],
       [[14, 15],
        [18, 19],
        [22, 23]]])

 

6、根据索引条件改变数组值的函数

 

　　np.select(condlist, choicelist, default=0)　　np.select方法根据条件返回符合列表条件的元素数组。condlist表示输入的条件列表，choicelist表示输入的选择列表，与condlist是一一对应的，当x中元素符合condlist中的第一个条件时，输出数组中就选择choicelist中第一个选择作为输出，如符合多个条件则选择第一个。default表示不符合condlist中所有条件时填充什么元素，默认为0，选x时填充原数组的元素。

>>> x = np.arange(10)
>>> condlist = [x<3, x>5]
>>> choicelist = [x, x**2]
>>> np.select(condlist, choicelist)
>>> np.select(condlist, choicelist,default=x)
array([ 0,  1,  2,  3,  4,  5, 36, 49, 64, 81])

>>> np.select(condlist, choicelist)
array([ 0,  1,  2,  0,  0,  0, 36, 49, 64, 81])

 

　　np.place(a, mask, b)　　np.place方法根据布尔数组mask的索引，将索引到的输入数组a的元素的值更改为b，如果b的长度不够则重复b。

>>> arr = np.arange(6).reshape(2, 3)
>>> np.place(arr, arr>2, [44, 55])
>>> arr
array([[ 0,  1,  2],
       [44, 55, 44]])

 

　　np.put(a, index, b)　　np.put方法将用给定值b替换输入数组a中索引为index的元素的值。index是按展平后的数组索引的，替换的值按index中的顺序排列b，如果b的长度不够则重复b。

>>> arr = np.arange(6).reshape(2, 3)
>>> np.put(arr,[0,1,5,4],[44,55])
>>> arr
array([[44, 55,  2],
       [ 3, 55, 44]])

 

 

参考：numpy中文文档：https://www.numpy.org.cn/reference/

　　　numpy英文文档：https://numpy.org/doc/1.17/reference/index.html

 

iwehdio的博客园：https://www.cnblogs.com/iwehdio/