numpy多维数组元素筛选

Posted 2023-02-16 修炼之路

导读

在python数据操作中我们经常会用到numpy，通过numpy来操作数组非常的方便和快速

元素筛选

• 一维数组
  筛选出数组中等于某个值的数组

import numpy as np

a = np.array([1,2,3,4,2,4,5])
#筛选出元素
print(a[a == 2])
#筛选出元素的下标
print(np.where(a == 2))

• 多维数组
  筛选的目标元素是一个多维数组，这时候我们就需要使用all和any函数

import numpy as np

a = np.array([[0,0],[1,1],[2,2],[0,0],[3,4],[1,5]])
#目标元素
target = np.array([0,0])
#筛选出元素
print(a[(a==target).all(axis=1)])

#目标元素
target = np.array([1,2])
#筛选出第一列包含1或者第二列包含2的元素
print(a[(a==target).any(axis=1)])