Python遍历数组的三种方法

在Python中，遍历数组是日常编程中必须操作之一。本文将介绍三种遍历数组的方法，分别为 for 循环、while 循环和 numpy.nditer() 方法。

1. for循环

for循环是Python中最基础的循环方式，同样适用于Python中的数组遍历。语法如下：

for element in array:
    # do something with element

其中 element 为数组中的元素， array 是要遍历的数组名称。示例代码如下：

arr = [1, 2, 3, 4, 5]
for num in arr:
    print(num)

执行代码，输出依次为：

1
2
3
4
5

代码已经很简洁了，就不再详细解释。需要注意的是，python中的for循环并不是传统的for循环，而是类似于Java中的for-each循环。

2. while循环

while循环同样适用于Python中的数组遍历。语法如下：

i = 0
while i < len(array):
    # do something with array[i]
    i += 1

其中 i 为数组下标，初始为0，同时 len() 函数用于计算数组的长度。示例代码如下：

arr = [1, 2, 3, 4, 5]
i = 0
while i < len(arr):
    print(arr[i])
    i += 1

执行代码，输出结果与for循环相似。需要注意的是，while循环需要手动累加数组下标，相比for循环略显麻烦。

3. numpy.nditer()方法

在Python中，可以使用numpy库中的 nditer() 方法来遍历数组，特别是对于多维数组，能够大大简化代码。使用 nditer() 方法时，需要注意通过 flags 参数设置遍历顺序，如正序遍历、倒序遍历等。示例代码如下：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
for num in np.nditer(arr):
    print(num, end=' ')

输出结果如下：

1 2 3 4 5 6

使用 nditer() 方法相比for循环和while循环更为简洁，且其对于多维数组的遍历效果更佳。

综上所述，这三种方法都是比较常用的Python遍历数组的方式，可以根据不同的需求进行选择使用。