numpy如何获取array中数组元素的索引位置

目录

• numpy - 获取array中数组元素的索引
• 1. 函数原型
• 2. 应用
• 2.1 一维数组
• 2. 2 二维数组

numpy - 获取array中数组元素的索引

    

                                            2017年08月05日 10:36:59
        

            阅读数：1797
                                        
    


    

                
        

1. 函数原型

argwhere(array)：找到非空数组array在满足某些条件下的索引，返回索引数组。

2. 应用

2.1 一维数组

返回一个一维数组，代表当前满足条件的元素出现的位置。

# -*- coding: utf-8 -*-  
import numpy as np  
  
arr = np.random.randint(0,10, (5,))  
index = np.argwhere(arr < 5)

# -*- coding: utf-8 -*-
import numpy as np

arr = np.random.randint(0,10, (5,))
index = np.argwhere(arr < 5)

2. 2 二维数组

返回二维数组，代表当前满足条件的元素出现的位置。

# -*- coding: utf-8 -*-  
import numpy as np  
  
”“” 
arr =  
    9 3 7 0  
    3 4 2 4  
    3 6 4 4  
     
index =  
    0   1 
    0   3 
    1   0 
    1   1 
    1   2 
    1   3 
    2   0 
    2   2 
    2   3 
”“”  
arr = np.random.randint(0,10, (3,4))  
index = np.argwhere(arr < 5)  

# -*- coding: utf-8 -*-
import numpy as np

"""
arr = 
    9 3 7 0 
    3 4 2 4 
    3 6 4 4 

index = 
    0   1
    0   3
    1   0
    1   1
    1   2
    1   3
    2   0
    2   2
    2   3
"""

arr = np.random.randint(0,10, (3,4))
index = np.argwhere(arr < 5)

参考文献

http://blog.csdn.net/vernice/article/details/50990919

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/ZK_J1994/article/details/76707734

文章标签： Python numpy

个人分类： Python

所属专栏： Python

到此这篇关于numpy获取array中数组元素的索引位置的文章就介绍到这了,更多相关numpy array数组索引内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！