numpy: np.where

Syntax

numpy.where(condition[, x, y])

返回 符合某一条件下标

Note : 不接受 list 型的参数,只接受 `ndarray 型输入。

用法

单目运算

import numpy as np
array_1 = np.array([1, 0, 1, 5, 5, 1])
array_2 = np.array([[1, 0, 1],
                    [5, 5, 1]])
print np.where(array_1 > 2)
print np.where(array_2 > 2)

打印结果:

(array([3, 4]),)

(array([1, 1]), array([0, 1]))

三目运算

例:将奇数转换为偶数,偶数转换为奇数:

import numpy as np
y = np.array([1, 2, 3, 4, 5, 6])
print np.where(y%2 == 0, y+1, y-1)

打印结果:

[0 3 2 5 4 7]


本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏小樱的经验随笔

洛谷 P1019 单词接龙【经典DFS,温习搜索】

P1019 单词接龙 题目描述 单词接龙是一个与我们经常玩的成语接龙相类似的游戏,现在我们已知一组单词,且给定一个开头的字母,要求出以这个字母开头的最长的“龙”...

3816
来自专栏数据结构与算法

P3391 文艺平衡树

hh 题目描述 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,例如原有序序列是5 4 3 2 1,翻转区间是[...

29810
来自专栏数据结构与算法

洛谷P3357 最长k可重线段集问题(费用流)

题目描述http://www.cnblogs.com/zwfymqz/p/8559566.html 给定平面 x-O-yx−O−y 上 nn 个开线段组成的集合...

4106
来自专栏我的博客

C编程笔记

1.编译命令gcc test.c -o test 带上参数o就是指定编译文件名 2.printf(“%.2lf”,b) 其中前面2是小数点后位数,l是字母...

3695
来自专栏前端儿

ASCII码排序

输入第一行输入一个数N,表示有N组测试数据。后面的N行输入多组数据,每组输入数据都是占一行,有三个字符组成,之间无空格。输出对于每组输入数据,输出一行,字符中间...

1872
来自专栏个人分享

二维数组中的查找

在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该...

1092
来自专栏黑泽君的专栏

java基础学习_基础语法(下)02_day06总结

============================================================================= ==...

611
来自专栏数据结构与算法

23:二维数组回形遍历

23:二维数组回形遍历 总时间限制: 1000ms 内存限制: 65536kB描述 给定一个row行col列的整数数组array,要求从array[0][0...

5086
来自专栏用户2442861的专栏

python 中 常用到的 numpy 函数 整理

http://blog.csdn.net/timidsmile/article/details/16963699

812
来自专栏机器学习之旅

tf.scan 记录

tf.scan(fn, elems, initializer=None, parallel_iterations=10, back_prop=True, swa...

1262

扫码关注云+社区

领取腾讯云代金券