python - 选择 Numpy 数组的一部分

Question

我是 Python 的新手，我阅读了一些有经验的人编写的代码。这部分应该是 Numpy 数组的一部分

a=np.random.random((10000,32,32,3)) # random values as an example
mask=list(range(5000))
a=a[mask]

对我来说，创建另一个列表来获取数组的一部分看起来相当浪费。此外，结果数组实际上是前 5000 个字段，不需要复杂的选择。

据我所知，以下代码应该给出相同的结果：

a=a[:5000]

第一个例子有什么好处？它更快吗？还是我错过了什么？

标签： pythonnumpyslice