python求数组某一列的维度

假设你有一个二维数组 `arr`，要求获取其第二列的维度，可以使用以下代码：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col_dimension = arr[:, 1].shape

print(col_dimension)

输出结果为 `(3,)`，表示第二列有 3 个元素。