当前位置:首页 » 数据分析 » 正文

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

看: 836次  时间:2021-01-20  分类 : 数据分析

举个例子:

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博客。

标签:pandas  numpy  

<< 上一篇 下一篇 >>

搜索

推荐资源

  Powered By python教程网   鲁ICP备18013710号
python博客 - 小白学python最友好的网站!