Python如何遍历numpy数组

更新时间：2022年11月08日 09:31:27 作者：cchangcs

这篇文章主要介绍了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来进行遍历。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python的字典和集合你了解吗
章主要为大家详细介绍了python的字典和集合，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2022-02-02
Python实现将Word和Excel文件转换为PPT
在日常工作中,我们经常需要将多个Word文档或Excel表格的内容汇总到一个PPT演示文稿中,手动执行这项任务可能非常耗时,因此,使用Python自动化这个过程可以大大提高效率,所以本文给大家介绍了Python实现将Word和Excel文件转换为PPT,需要的朋友可以参考下
2024-08-08
python-xpath获取html文档的部分内容
这篇文章主要介绍了python-xpath获取html文档的部分内容，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-03-03
python实现的config文件读写功能示例
这篇文章主要介绍了python实现的config文件读写功能,结合实例形式分析了Python文件读写相关操作技巧,需要的朋友可以参考下
2019-09-09
Python 初始化多维数组代码
python多维数组方法示例代码
2008-09-09
Python处理HTTP认证的常见方法
在Python中,HTTP认证通常指的是客户端在向服务器发送请求时,需要提供某种形式的认证信息,处理HTTP认证通常涉及到使用requests库,requests库提供了简单的方式来处理需要认证的HTTP请求,本文给大家介绍了Python处理HTTP认证的常见方法
2025-03-03
Python Matplotlib基本用法详解
Matplotlib 是Python中类似 MATLAB 的绘图工具，熟悉 MATLAB 也可以很快的上手 Matplotlib，这篇文章主要介绍了Python Matplotlib基本用法,需要的朋友可以参考下
2023-03-03
Python3实现的判断回文链表算法示例
这篇文章主要介绍了Python3实现的判断回文链表算法,结合实例形式分析了Python3针对链表是否为回文链表进行判断的相关算法实现技巧,需要的朋友可以参考下
2019-03-03
Python中sorted()用法案例代码
sorted() 作为 Python 内置函数之一，其功能是对序列（列表、元组、字典、集合、还包括字符串）进行排序，这篇文章主要介绍了Python中sorted()用法,需要的朋友可以参考下
2023-02-02
python实现实时视频流播放代码实例
这篇文章主要介绍了python实现实时视频流播放代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-01-01