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代码，有一个数组和一个csv文件，数组对应一个column，要利用数组作为索引，挑选出csv中的数据，要挑选出符合的行

以下是一段示例代码，可以根据你的具体需求进行修改：

import csv

# 假设数组为 [1, 3, 5]，对应 csv 文件的第二列
array = [1, 3, 5]

# 打开 csv 文件
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    # 遍历每一行数据
    for row in reader:
        # 如果当前行的第二列在数组中，则输出该行数据
        if int(row[1]) in array:
            print(row)

注意，这段代码仅供参考，具体实现方式可能因为数据格式等原因而有所不同。