美文网首页
NumPy中的花式索引与np.ix_函数

NumPy中的花式索引与np.ix_函数

作者: glRu | 来源:发表于2020-06-19 12:46 被阅读0次

参考文章:

【1】https://www.numpy.org.cn/user/quickstart.html#%E8%8A%B1%E5%BC%8F%E7%B4%A2%E5%BC%95%E5%92%8C%E7%B4%A2%E5%BC%95%E6%8A%80%E5%B7%A7

【2】https://blog.csdn.net/weixin_40001181/article/details/79775792

用法与例子如下:

创建 arr 数组

>>> arr1 = np.empty((8,4))        # 创建一个8行4列的二维数组

>>> for i in range(8):           # 每一行赋值为0~7

    arr1[i] = i

>>> arr1

array([[ 0.,  0.,  0.,  0.],

      [ 1.,  1.,  1.,  1.],

      [ 2.,  2.,  2.,  2.],

      [ 3.,  3.,  3.,  3.],

      [ 4.,  4.,  4.,  4.],

      [ 5.,  5.,  5.,  5.],

      [ 6.,  6.,  6.,  6.],

      [ 7.,  7.,  7.,  7.]])

1、用1维数组进行索引

2、用有负数的1维数组进行索引,就是从末尾开始选取行

新建一个数组 arr2 

>>> arr2 = np.arange(32).reshape((8,4))

>>> arr2

array([[ 0,  1,  2,  3],

      [ 4,  5,  6,  7],

      [ 8,  9, 10, 11],

      [12, 13, 14, 15],

      [16, 17, 18, 19],

      [20, 21, 22, 23],

      [24, 25, 26, 27],

      [28, 29, 30, 31]])

3、按坐标选取每一个数

4、希望先按我们要求选取行,再按顺序将列排序,获得一个矩形

先按先选取第1、5、2、7行,每一行再按第0个、第3个、第1个、第2个排序

5、np.ix_函数,能把两个一维数组 转换为 一个用于选取方形区域的索引器

实际意思就是,直接往np.ix_()里扔进两个一维数组[1,5,7,2],[0,3,1,2],就能先按我们要求选取行,再按顺序将列排序,跟上面得到的结果一样,而不用写“[ : , [0,3,1,2] ]”

原理:np.ix_函数就是输入两个数组,产生笛卡尔积的映射关系

例如就这个例子,np.ix_函数,将数组[1,5,7,2]和数组[0,3,1,2]产生笛卡尔积,就是得到(1,0),(1,3),(1,1),(1,2);(5,0),(5,3),(5,1),(5,2);(7,0),(7,3),(7,1),(7,2);(2,0),(2,3),(2,1),(2,2);然后按照对应的索引取得对应的值。

相关文章

网友评论

      本文标题:NumPy中的花式索引与np.ix_函数

      本文链接:https://www.haomeiwen.com/subject/crevxktx.html