如何从numPy数组中提取一组给定的列?

时间:2013-12-20 06:17:46

标签: python numpy

给出一个numpy数组,例如:

x = array([[0,  1,  2,  3],
          [ 4,  5,  6,  7],
          [ 8,  9, 10, 11],
          [12, 13, 14, 15]])

如何形成由第一列和第三列组成的新数组?

1 个答案:

答案 0 :(得分:2)

要从数组中提取第一列和第三列,请使用以下语法:

X [:,[0,2]]

这意味着,取所有行,仅选择第0列和第2列。请注意,numPy数组中的索引从零开始。