Python如何遍历numpy数组
作者:cchangcs 发布时间:2023-04-15 00:40:17
标签:Python,遍历,numpy,数组
Python遍历numpy数组
下面是示例代码:
import numpy as np
num = np.zeros([2, 3])
[rows, cols] = num.shape
print(rows, cols)
for i in range(rows):
for j in range(cols):
print(num[i, j])
运行结果:
2 3
0.0
0.0
0.0
0.0
0.0
0.0
numpy中数组的遍历技巧分享
在numpy中,当需要循环处理数组中的元素时,能用内置通函数实现的肯定首选通函数,只有当没有可用的通函数的情况下,再来手动进行遍历,遍历的方法有以下几种
1. 内置for循环
最基础的遍历方法还是for循环,用法如下
# 一维数组,和普通的python序列对象一致
>>> a
array([0, 1, 2, 3, 4])
>>> for i in a:
... print(i)
...
0
1
2
3
4
# 二维数组,每次遍历一行,以列表的形式返回一行的元素
>>> a = np.arange(12).reshape(3, 4)
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> for i in a:
... print(i)
...
[0 1 2 3]
[4 5 6 7]
[ 8 9 10 11]
for循环中得到的是对应元素的副本,所以通过上述方式只能访问,不能修改原始数组中的值。
2. flat迭代器
数组的flat属性返回的是数组的迭代器,通过这个迭代器,可以一层for循环就搞定多维数组的访问,用法如下
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> for i in a.flat:
... print(i)
...
0
1
2
3
4
5
6
7
8
9
10
11
3. nditer迭代器
numpy中的nditer函数可以返回数组的迭代器,该迭代器的功能比flat更加强大和灵活,在遍历多维数组时,通过order参数可以指定遍历的顺序,C表示C语言的风格,优先处理行,F表示Fortran语言的风格,优先处理列,用法如下
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
# 默认按行处理
>>> for i in np.nditer(a, order='C'):
... print(i)
...
0
1
2
3
4
5
6
7
8
9
10
11
# 按列处理
>>> for i in np.nditer(a, order='F'):
... print(i)
...
0
4
8
1
5
9
2
6
10
3
7
11
普通的遍历只能访问元素,而nditer可以允许我们在遍历的同时修改原始数组中的元素,只需要op_flags参数即可,用法如下
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> for i in np.nditer(a, op_flags = ['readwrite']):
... i *= 2
...
>>> a
array([[ 0, 2, 4, 6],
[ 8, 10, 12, 14],
[16, 18, 20, 22]])
>>> for i in np.nditer(a, op_flags = ['writeonly']):
... i += 2
...
>>> a
array([[ 2, 4, 6, 8],
[10, 12, 14, 16],
[18, 20, 22, 24]])
nditer更强大的功能在于广播遍历,通过内置的广播机制,可以实现两个数组的组合,用法如下
>>> a = np.arange(12).reshape(3, 4)
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> b = np.arange(4)
>>> b
array([0, 1, 2, 3])
>>> np.nditer([a, b])
<numpy.nditer object at 0x7f9db6b11170>
>>> for x,y in np.nditer([a,b]):
... print(x,y)
...
0 0
1 1
2 2
3 3
4 0
5 1
6 2
7 3
8 0
9 1
10 2
11 3
```
```
>>> b = np.arange(3).reshape(-1, 1)
>>> b
array([[0],
[1],
[2]])
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
>>> for x,y in np.nditer([a,b]):
... print(x,y)
...
0 0
1 0
2 0
3 0
4 1
5 1
6 1
7 1
8 2
9 2
10 2
11 2
简单的元素访问直接使用for循环迭代数组即可,注意二维数组和一维数组的区别,nditer的3个特点对应不同的使用场景,当遇到对应的情况时,可以选择nditer来进行遍历。
来源:https://cchang.blog.csdn.net/article/details/79109843


猜你喜欢
- 在国内利用Python从Internet上爬取数据时,有些网站或API接口被限速或屏蔽,这时使用代理可以加速爬取过程,减少请求失败,Pyth
- 朋友需要一个工具,将图片中的文字提取出来。我帮他在网上找了一些OCR的应用,都不好用。所以准备自己研究,写一个Web APP供他使用。OCR
- 登录百度AL开发平台在控制台选择语音合成创建应用填写应用信息在应用列表获取(Appid、API Key、Secret Key)6. 安装py
- JOSN字符串转换为自定义类实例对象有时候我们有这种需求就是把一个JSON字符串转换为一个具体的Python类的实例,比如你接收到这样一个J
- 本文实例为大家分享了JS实现轮播图特效的具体代码,供大家参考,具体内容如下知识点轮播图思想:① 建立一个全局变量索引,始终标记当前显示图片。
- 1.auto close tagHTML自动补全标签2.beautiful UI32个主题集合,具体使用看个人喜好。3.better com
- 程序编制 随着WEB的发展,与浏览者的交互越来越受到网站制作者的重视。现在的主页跟最初的静态主
- 1. sys_getloadavg()sys_getloadavt()可以获得系 统负载情况。该函数返回一个包含三个元素的数组,每个元素分别
- 用js实现一个砖头在页面,但鼠标点击拖动时,砖头在页面上形成拖拉效果:刚开始时:鼠标点击拖动后:实现代码:<html> <
- 本文分享了php结合ajax实现无刷新上传图片的实例代码,分享给大家,希望大家可以和小编一起学习学习,共同进步。1.引入文件<!--图
- 在python中安装非自带python模块,有三种方式:1.easy_install2.pip3.下载压缩包(.zip, .tar, .ta
- 目的:方便调试,查看中间结果,因为觉得设断点调试相对麻烦。【运行环境:macOS 10.13.3,PyCharm 2017.2.4】老手:选
- 项目技术:webpack + vue + element + axois (vue-resource) + less-loader+ ...
- 记得导入包,其他按键可类比def keyPressEvent(self, event): if event.key() == Q
- 概要 简单介绍几种用于判断numpy数组是否全
- 本文实现利用python的socketserver这个强大的模块实现套接字的并发,具体内容如下目录结构如下:测试文件请放在server_fi
- 本文实例讲述了Python设计模式之简单工厂模式。分享给大家供大家参考,具体如下:简单工厂模式(Simple Factory Pattern
- 有感于中国互联网设计界十几年的变化,从网页设计师变身界面设计师,和近一两年来兴起的转型交互设计师。大多数都是随着一个行业的兴起而前赴后继的投
- python窃取摄像头照片源码+获取授权码方法+py打包成exe教你用python做一个属于自己的窃取摄像头照片的软件。需要安装python
- 本文实例为大家分享了雪花飞舞效果javascript实现,供大家参考,具体内容如下原理:1、js动态创建DIV,指定CLASS类设置不同的背