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]`表示获取所有行的第三列。