Numpy np.where()的简单用法

第一种用法是寻找符合条件的数的索引：

import numpy as nparr = np.array([1, 2, 3, 4, 5])
idx = np.where(arr > 3)
print(idx)

输出：

(array([3, 4], dtype=int64),)

第二种用法是寻找符合条件的数并进行修改：

import numpy as nparr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 3, 1, 0)
print(new_arr)

输出：

[0 0 0 1 1]

其实就相当于第一种方法拿到索引后再修改：

import numpy as nparr = np.array([1, 2, 3, 4, 5])
idx1 = np.where(arr > 3)
idx2 = np.where(arr <= 3)
arr[idx1[0]] = 1
arr[idx2[0]] = 0
print(arr)

还有其他的用法，不过本文不再列出。

Numpy np.where()的简单用法相关推荐

python matplotlib 简单用法
python matplotlib 简单用法具体内容请参考官网代码 import matplotlib.pyplot as plt import numpy as np # 支持中文 plt.rc ...
笔记：numpy array中冒号的用法
在未引包的情况下,python中的冒号的用法比较简单,诸如a[10:20], 或者a[::10] 而在引入numpy后,冒号的用法得到了扩充,可以用引号来表示一整个维度,从开始到结束的所有元素,例如: ...
numpy中where函数的用法
numpy中where函数的用法 numpy.where(condition,x,y)--若满足condition,输出x,否则输出y. 举例: 例1:一维数组 >>> a = np ...
python numpy np.lexsort()（使用键序列执行间接稳定排序）（具体没太搞懂区别？）
from numpy\core_multiarray_umath.py @array_function_from_c_func_and_dispatcher(_multiarray_umath.lex ...
python画函数图像要用到的模块_教你如何绘制数学函数图像——numpy和matplotlib的简单应用...
numpy和matplotlib的简单应用一.numpy库 1.什么是numpy NumPy系统是Python的一种开源的数值计算扩展.这种工具可用来存储和处理大型矩阵,比Python自身的嵌套列表 ...
python中numpy的用法_Python中numpy多维数组的用法
继上篇讲过numpy如何构建多维数组之后,今天我们来学习numpy多维数组的用法. 加法和减法操作要求操作双方的维数信息一致,均为M*N为数组方可正确执行操作. a = np.arange(4) 输出 ...
python count函数代码_python中count函数简单用法
python中count函数简单用法 python中count函数的用法 Python count()方法描述 Python count() 方法用于统计字符串里某个字符出现的次数.可选参数为在字符 ...
Numpy和Pandas的简单使用
Numpy和Pandas的简单使用 Numpy Numpy基本数据结构 Numpy基本操作和属性 Numpy矩阵运算 Pandas pandas数据读取 pandas数据展示 Pandas数据处理 P ...
Numpy np.max和np.maximum实现relu函数
Numpy np.max和np.maximum实现relu函数 1. np.max 计算一个数组中的最值 1. np.max 计算一个数组中的最值 (1)基础用法 a = np.array([[1, ...

Numpy np.where()的简单用法

Numpy np.where()的简单用法相关推荐

最新文章

热门文章