Python查找指定元素的索引（bool索引）

原创已于 2022-11-23 00:41:24 修改 · 2.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #开发语言

于 2022-11-22 22:16:10 首次发布

Python 专栏收录该内容

47 篇文章

订阅专栏

本文通过四个具体示例介绍如何使用Python的NumPy库进行数组元素查找、赋值及统计等操作，包括不同类型的变量赋值方式对数组的影响。

看下面的几个例子，其中例1、例2结合了变量赋值传递时的传值、传址、深拷贝情况：

例1：查找指定元素的索引 + 赋值

【注】：下面用 c = a （属于传址、引用、传可变对象），改变 c 也改变原变量 a 的值。

import numpy as np

a = np.random.randint(-2, 2, (1, 10))
print ("随机数组a(前)  :  "+str(a))  #输出数组a(前)
c = a
c[(a==0)] = 999              #条件为a==0
print ("改变指定数值后:  "+str(c))  #输出满足条件的个数
print ("随机数组a(后)  :  "+str(a))  #输出数组a(后)

结果：

随机数组a(前)  :  [[ 1  0 -1 -1  1  0  0  1  0 -1]]
改变指定数值后:  [[  1 999  -1  -1   1 999 999   1 999  -1]]
随机数组a(后)  :  [[  1 999  -1  -1   1 999 999   1 999  -1]]

例2：查找指定元素的索引 + 赋值

【注】：下面用 c = copy.deepcopy(a) （属于传值、深拷贝），改变 c 不改变原变量 a 的值。

import numpy as np
import copy

a = np.random.randint(-2, 2, (1, 10))
print ("随机数组a(前)  :  "+str(a))  #输出数组a(前)
c = copy.deepcopy(a)
c[(a==0)] = 999              #条件为a==0
print ("改变指定数值后:  "+str(c))  #输出满足条件的个数
print ("随机数组a(后)  :  "+str(a))  #输出数组a(后)

结果：

随机数组a(前)  :  [[ 0 -2 -1 -1 -2  1  0 -1 -2 -1]]
改变指定数值后:  [[999  -2  -1  -1  -2   1 999  -1  -2  -1]]
随机数组a(后)  :  [[ 0 -2 -1 -1 -2  1  0 -1 -2 -1]]

例3：查找指定元素的索引 + 统计个数

import numpy as np

a = np.random.randint(-2, 2, (1, 10))
c=np.sum(a>=1)              #条件为a>=1
print ("随机数组a  :  "+str(a))  #输出数组a
print ("大于等于1的个数:  "+str(c))  #输出满足条件的个数

结果：

随机数组a  :  [[-1  0  1  1 -2 -2  1  1  0  1]]
大于等于1的个数:  5

例4：查找指定元素的索引 + 统计个数

import numpy as np

a = np.random.randint(-2, 2, (1, 10))
c=np.sum(a==0)              #条件为a==0
print ("随机数组a  :  "+str(a))  #输出数组a
print ("等于0的个数:  "+str(c))  #输出满足条件的个数

结果：

随机数组a  :  [[ 0  0 -2 -2 -1 -2  0 -1 -2  0]]
等于0的个数:  4