以下是一个示例代码,用于遍历二维数组的列作为列向量:
import numpy as np
# 创建一个二维数组
array = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 获取数组的列数
num_cols = array.shape[1]
# 遍历数组的列
for col_index in range(num_cols):
# 获取当前列向量
column_vector = array[:, col_index]
print("第", col_index+1, "列向量:", column_vector)
输出示例:
第 1 列向量: [1 4 7]
第 2 列向量: [2 5 8]
第 3 列向量: [3 6 9]
该示例使用了NumPy库来创建和操作数组。首先,我们创建了一个二维数组array
。然后,使用shape
属性获取数组的列数num_cols
。接下来,使用一个循环遍历从0到num_cols-1
的列索引。在每次循环中,使用切片操作array[:, col_index]
获取当前列的列向量,并将其打印输出。
上一篇:遍历数组的列以标准化数据