在 numpy 中,可以使用 numpy.nditer
对数组的每一列进行遍历。
示例代码如下:
import numpy as np
# 创建2x3 的数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 对每一列遍历
for col in np.nditer(arr, flags=['external_loop'], order='F'):
print(col)
在这段代码中,flags
参数设置为 'external_loop'
表示对每一列进行遍历,order
参数设置为 'F'
表示以列序遍历。
注意,如果输出 col 是循环遍历列中的每个元素,而不是整列