Python 获取numpy.array索引值的实例

yipeiwu_com5年前Python基础

举个例子:

q=[0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]

我想获取其中值等于7的那个值的下标,以便于用于其他计算。

如果使用np.where,如:

q=np.arange(0,16,1)

g=np.where(q==7)

print q

print g

运行结果是:

[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]

(array([7]),)

显然(array([7]),)中的数字7我是没法提取出来做运算的,这是一个tuple。

处理方法是:

q=np.arange(0,16,1)

g=np.argwhere(q==7)//注意这里,返回的是下标而不是索引值

print q

print g

print int(g)//转化为可运算的数字

运行结果为:

[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]

[[7]]

7

以上这篇Python 获取numpy.array索引值的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

用openCV和Python 实现图片对比,并标识出不同点的方式

用openCV和Python 实现图片对比,并标识出不同点的方式

最近项目中需要实现两组图片对比,并能将两者的区别标识出来。 在网上搜索一大堆找到一篇大神的文章,最终实现该功能,在这里记录下: 想要实现此demo,首先我们得确保电脑上已安装 openC...

Python动态导入模块的方法实例分析

本文实例讲述了Python动态导入模块的方法。分享给大家供大家参考,具体如下: 一、正常导入模块 正常模块导入方式: import module(模块路径) 同时导入多个模块: im...

pytorch点乘与叉乘示例讲解

点乘 import torch x = torch.tensor([[3,3],[3,3]]) y = x*x #x.dot(x) z = torch.mul(x,x) #x.mul...

python整小时 整天时间戳获取算法示例

根据当前时间戳获得整小时时间戳 unit = 3600 start_time = int(time.time())/3600 * 3600 根据当前时间戳获得整天时间戳 uni...

python处理xml文件的方法小结

本文实例讲述了python处理xml文件的方法。分享给大家供大家参考,具体如下: 前一段时间因为工作的需要,学习了一点用Python处理xml文件的方法,现在贴出来,供大家参考。 xml...