NumPy索引与切片的用法示例总结
作者:zhenguo 时间:2022-05-13 19:33:38
前言
索引和切片是NumPy中最重要最常用的操作。熟练使用NumPy切片操作是数据处理和机器学习的前提,所以一定要掌握好。
参考NumPy官方文档,总结NumPy索引和切片,可以看到它们相比Python更加方便、简介和强大。
索引和切片
您可以使用与切片 Python列表相同的方法,对NumPy数组进行索引和切片。
>>> data = np.array([1, 2, 3])
>>> data[1]
2
>>> data[0:2]
array([1, 2])
>>> data[1:]
array([2, 3])
>>> data[-2:]
array([2, 3])
你可以这样想象:
您可能需要获取数组的一部分或特定数组元素,以便在进一步分析或其他操作中使用。为此,需要对数组进行子集、切片和/或索引。
如果您想从数组中选择满足特定条件的值,那么NumPy很简单。
例如,如果从这个数组开始:
>>> a = np.array([[1 , 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
可以轻松打印数组中小于5的所有值。
>>> print(a[a < 5])
[1 2 3 4]
例如,还可以选择等于或大于5的数字,并使用该条件对数组进行索引。
>>> five_up = (a >= 5)
>>> print(a[five_up])
[ 5 6 7 8 9 10 11 12]
可以选择可被2整除的元素:
>>> divisible_by_2 = a[a%2==0]
>>> print(divisible_by_2)
[ 2 4 6 8 10 12]
或者可以使用&和|运算符选择满足两个条件的元素:
>>> c = a[(a > 2) & (a < 11)]
>>> print(c)
[ 3 4 5 6 7 8 9 10]
还可以使用逻辑运算符&和 |返回布尔值,指定数组中的值是否满足特定条件。这对于包含名称或其他分类值的数组很有用。
>>> five_up = (a > 5) | (a == 5)
>>> print(five_up)
[[False False False False]
[ True True True True]
[ True True True True]]
还可以使用np.nonzero()从数组中选择元素或索引。
从这个数组开始:
>>> a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
可以使用np.nonzero()打印元素的索引,例如,小于5:
>>> b = np.nonzero(a < 5)
>>> print(b)
(array([0, 0, 0, 0]), array([0, 1, 2, 3]))
在本例中,返回了一个数组元组:每个维度一个。第一个数组表示找到这些值的行索引,第二个数组表示找到这些值的列索引。
如果要生成元素所在的坐标列表,可以压缩数组,遍历坐标列表,然后打印它们。例如:
>>> list_of_coordinates= list(zip(b[0], b[1]))
>>> for coord in list_of_coordinates:
... print(coord)
(0, 0)
(0, 1)
(0, 2)
(0, 3)
还可以使用np.nonzero()打印数组中小于5的元素,并使用:
>>> print(a[b])
[1 2 3 4]
如果要查找的元素在数组中不存在,则返回的索引数组将为空。例如:
>>> not_there = np.nonzero(a == 42)
>>> print(not_there)
(array([], dtype=int64), array([], dtype=int64))
总结
来源:https://mp.weixin.qq.com/s/QBDDxPVzBQalPJKH7oNiow
标签:numpy,索引,切片
0
投稿
猜你喜欢
数据库Left join , Right Join, Inner Join 的相关内容,非常实用
2024-01-16 16:59:56
详解VScode自动补全CSS3前缀插件以及配置无效的解决办法
2023-01-05 06:49:40
如何解决cmd运行python提示不是内部命令
2022-02-05 05:05:38
Python使用requests发送POST请求实例代码
2022-10-26 07:40:00
Python使用tkinter库实现文本显示用户输入功能示例
2023-09-21 11:00:17
关闭时刷新父窗口两种方法
2024-06-11 20:21:24
Python中如何获取类属性的列表
2023-02-10 21:57:41
一文学会利用python解决文章付费限制问题
2021-04-09 08:23:51
python爬虫中抓取指数的实例讲解
2022-02-18 04:59:58
python将人民币转换大写的脚本代码
2021-03-05 17:36:23
TensorFlow人工智能学习按索引取数据及维度变换详解
2024-01-04 13:18:55
无忧贴子管理器(ListView组件)
2009-01-02 17:56:00
python中defaultdict用法实例详解
2022-08-09 17:01:10
Vscode常用快捷键列表、插件安装、console.log详解
2023-02-11 01:29:04
如何远程使用服务器上的Jupyter notebook
2023-02-08 10:45:54
Python pkg_resources模块动态加载插件实例分析
2023-10-31 16:48:39
Python自动重新加载模块详解(autoreload module)
2023-09-19 04:36:12
.NET Framework SQL Server 数据提供程序连接池
2024-01-27 05:05:23
解决layer弹出层msg的文字不显示的问题
2024-04-23 09:07:51
python数据爬下来保存的位置
2021-01-29 20:49:19