python array挑选第一列和第三列

可以使用 Python 的切片（slice）操作来挑选二维数组（或列表）的某些列。假设你有一个名为 `my_array` 的二维数组，其中第一列是索引为 0 的列，第二列是索引为 1 的列，第三列是索引为 2 的列，以此类推。那么要挑选第一列和第三列，可以使用以下代码：

# 示例二维数组
my_array = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 挑选第一列和第三列
col1_and_3 = [[row[0], row[2]] for row in my_array]
print(col1_and_3) # 输出: [[1, 3], [4, 6], [7, 9]]

上述代码中，我们使用列表推导式（list comprehension）来遍历二维数组的每一行，并将每行的第一列和第三列组成一个新的列表。这样，`col1_and_3` 就是一个只包含第一列和第三列的二维数组。

相关问题

python获取numpy第一列和第三列

假设有一个名为`arr`的numpy数组，可以通过以下代码获取第一列和第三列：

import numpy as np

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

# 获取第一列和第三列
col1 = arr[:, 0]
col3 = arr[:, 2]

print(col1)  # 输出 [1 4]
print(col3)  # 输出 [3 6]

在上面的代码中，`[:, 0]`表示获取所有行的第一列，而`[:, 2]`表示获取所有行的第三列。

表示array第一列并生成代码python

你可以使用Python中的列表(list)来表示数组(array)，并使用索引来访问它的第一列。在Python中，列表的索引从0开始，因此第一列可以表示为list的第0个元素。以下是一个生成Python代码的示例：

array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
first_column = [row[0] for row in array]
print(first_column)

输出：

[1, 4, 7]

以上Python代码实现了从二维数组中提取第一列(即所有子数组的第0个元素)。您可以根据需要修改数组和列索引。