python如何迭代数组

2019-01-24 阅读量: 589

数据分析师 Python数据分析

python如何迭代数组

扫码加入数据分析学习群

NumPy包中包含一个迭代器对象numpy.nditer。它是一个有效的多维迭代器对象，使用它可以迭代数组。使用Python的标准Iterator接口访问数组的每个元素。

# Python program for
# iterating over array

import numpy as geek

# creating an array using arrange
# method
a = geek.arange(12)

# shape array with 3 rows and
# 4 columns
a = a.reshape(3,4)

print('Original array is:')
print(a)
print()

print('Modified array is:')

# iterating an array
for x in geek.nditer(a):
print(x)