选择Numpy数组的一部分

时间:2018-11-02 15:48:31

标签: python numpy slice

我是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]

第一个示例的优点是什么?它更快吗?还是我错过了什么?

0 个答案:

没有答案