栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

numpy的函数np.where详细讲解。

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

numpy的函数np.where详细讲解。

numpy里有一个非常神奇的函数叫做np.where()函数,下面我们来讲解一下这个函数的用法。
主要有两个用法。

1. np.where(condition,x,y)

首先传入y,然后如果满足条件condition,那么y中原来的数值变为x,否则不变化。
相当于[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]
用法如下:

import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
arr1 = np.where(arr % 2 == 1, '奇数', arr)
print(arr1)

输出:

['0' '奇数' '2' '奇数' '4' '奇数' '6' '奇数' '8' '奇数' '10']
2. np.where(condition)

返回满足condition的索引。
用法如下:

import numpy as np
arr = np.array([11, 3, 4, 5, 6, 7, 8, 9])
print(np.where(arr < 5))

输出:

(array([1, 2], dtype=int64),)
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/744092.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号