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

yipeiwu_com6年前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设计】。

相关文章

Django 过滤器汇总及自定义过滤器使用详解

Django 过滤器 过滤器 描述 示例 upper 以大写方式输出 {...

PyTorch 解决Dataset和Dataloader遇到的问题

今天在使用PyTorch中Dataset遇到了一个问题。先看代码 class psDataset(Dataset): def __init__(self, x, y, trans...

Django学习教程之静态文件的调用详解

前言 静态文件是指 网站中的 js, css, 图片,视频等文件,本文主要给大家介绍了关于Django学习之静态文件调用的相关内容,分享出来供大家参考学习,下面话不多说了,来一起看看详细...

Pandas 解决dataframe的一列进行向下顺移问题

最近做比赛,有时候需要造出新的特征,而这次遇到的问题是将一列数据往下顺移一位。同时将开头缺失的那一个数据用其他方式填充。 df['feature'].shift(1)向下顺移一位,这时第...

用python画一只可爱的皮卡丘实例

用python画一只可爱的皮卡丘实例

效果图 #!/usr/bin/env python # -*- coding:utf-8 -*- from turtle import * ''' 绘制皮卡丘头部 ''' de...