python如何取出二维数组的某一列

在Python中，可以使用循环遍历二维数组并取出某一列的元素，也可以使用numpy库中的切片操作来实现。

假设有一个二维数组arr，要取出其中第二列的元素，可以使用以下两种方法：

方法一：循环遍历

for i in range(len(arr)):
    print(arr[i][1])

方法二：使用切片操作

import numpy as np
col = arr[:, 1]
print(col)

以上两种方法都可以取出二维数组的某一列。方法一是比较基础的方法，适用于所有情况；方法二使用了numpy库，适用于较大的数组，可以提高运行效率。

相关问题

python 给二维数组某一列后面添加数据

要在二维数组中的某一列后面添加数据，可以使用 NumPy 的切片操作和 `numpy.insert()` 函数。假设原数组为 `a`，我们想在第二列后面添加一组数据 `[10, 11, 12]`，可以按照以下步骤操作：

import numpy as np

# 创建一个 3x3 的二维数组
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 创建要添加的数据
new_data = np.array([10, 11, 12])

# 取出第二列之前的所有数据
head = a[:, :1]

# 取出第二列之后的所有数据
tail = a[:, 1:]

# 将新数据添加到第二列之后
new_col = np.insert(tail, 1, new_data, axis=1)

# 将头部数据和新数据拼接起来
new_array = np.concatenate((head, new_col), axis=1)

print(new_array)

输出结果如下所示：

[[ 1 10  2  3]
 [ 4 11  5  6]
 [ 7 12  8  9]]

在这个示例中，我们首先创建了要添加的数据 `new_data`，然后使用切片操作将原数组 `a` 分成了两部分：第二列之前的数据和第二列之后的数据。接着，我们使用 `np.insert()` 函数将新数据插入到第二列之后，得到了新的二维数组 `new_col`。最后，我们使用 `np.concatenate()` 函数将头部数据和新数据拼接起来，得到了最终的结果 `new_array`。

需要注意的是，切片操作得到的是原数组的一个视图，所以对切片操作返回的数组进行修改会直接影响到原数组。在这个示例中，我们对 `head` 数组和 `tail` 数组进行了修改，但是并没有影响到原数组 `a`。

python二维数组取某一列

可以使用列表推导式或numpy库的方法进行操作。假设二维数组为arr，要取第i列，可以使用下面两种方法：

1. 列表推导式

col = [arr[j][i] for j in range(len(arr))]

2. numpy库

import numpy as np

col = np.array(arr)[:, i].tolist()

这两种方法都可以让你得到一个列表，其中存储了二维数组中第i列的所有元素。