python - 关于 python 中索引 numpy 数组的澄清

Question

我是一名高中生，在查看一些 numpy 代码时，我发现了一些类似的内容

a = x[:,0:4]

x 是一个二维数组。我知道 a[:] 引用数组 a 中的所有对象，因此对于 x[:,0:4] ，它是否引用 x 的所有行和索引为 0,1,2,3 的列(不包括索引列) 4？

只是想确认它是如何工作的，因为我在几种类型的代码中看到过它，只是想确定一下。

Answer 1

你是对的。此 a = x[:,0:4] 选择前四列。

示例:

>>> a = np.arange(25).reshape(5, 5)
>>> a
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

您可以跳过 0，因为 a[:,:4] 与 a[:,0:4] 含义相同:

>>> a[:,:4]
array([[ 0,  1,  2,  3],
       [ 5,  6,  7,  8],
       [10, 11, 12, 13],
       [15, 16, 17, 18],
       [20, 21, 22, 23]])

你总是可以这样想:“第一维度第一，第二维度第二，依此类推。”在二维情况下，第一个维度是行，第二个维度是列。