python取某几列

在Python中，可以使用切片操作来选择某几列。假设有一个二维列表或NumPy数组，可以使用索引操作符（[]）来选择列。

下面是一些示例代码，演示如何选择某几列：

1. 对于二维列表：

# 假设有一个二维列表data
data = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12]
]

# 选择第1列和第3列
selected_columns = [0, 2]
result = [[row[col] for col in selected_columns] for row in data]
print(result)

输出：

[[1, 3], [5, 7], [9, 11]]

1. 对于NumPy数组：

import numpy as np

# 假设有一个NumPy数组arr
arr = np.array([
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12]
])

# 选择第1列和第3列
selected_columns = [0, 2]
result = arr[:, selected_columns]
print(result)

输出：

[[ 1  3]
 [ 5  7]
 [ 9 11]]

无论使用二维列表还是NumPy数组，选择某几列的方法都是类似的。只需提供要选择的列的索引列表，并使用切片操作进行选择。