python获取数组的某一列

可以使用列表推导式或numpy库来获取数组的某一列。

1. 使用列表推导式：

my_array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
col = [row[1] for row in my_array]  # 获取第2列
print(col)  # [2, 5, 8]

2. 使用numpy库：

import numpy as np

my_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col = my_array[:, 1]  # 获取第2列
print(col)  # [2 5 8]

相关问题

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 个元素。

python获取数组列数

在Python中，可以使用NumPy库来获取数组的列数。NumPy是一个用于科学计算的强大库，提供了许多用于数组操作的函数和方法。

要获取数组的列数，可以使用NumPy中的shape属性。shape属性返回一个元组，其中包含数组的维度信息。对于二维数组，元组的第二个元素即为列数。

下面是一个示例代码：

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 获取数组的列数
num_cols = arr.shape

print("数组的列数为:", num_cols)

输出结果为：

数组的列数为: 3