要从NumPy数组中提取特定列,可以使用切片操作。
以下是一个示例代码:
import numpy as np
# 创建一个3x3的NumPy数组
array = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 提取第一列
col1 = array[:, 0]
# 提取第二列
col2 = array[:, 1]
# 打印结果
print("第一列:", col1)
print("第二列:", col2)
输出结果:
第一列: [1 4 7]
第二列: [2 5 8]
在这个示例中,我们使用切片操作 [:, 0]
提取了第一列,使用切片操作 [:, 1]
提取了第二列。
上一篇:按条件从列表中删除元素
下一篇:按条件的Oracle SUM