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

Python  /  管理员 发布于 7年前   339

举个例子：

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 qprint 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 qprint gprint int(g)//转化为可运算的数字

运行结果为：

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

以上这篇Python 获取numpy.array索引值的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。