我只需要获取数组的小数部分.
使用numpy或简单地使用python modf函数很方便.
如果我们有大量正分数数据数组,例如,可以和(1000000,3)一样大,那么更方便的做法是:
numpy.modf(array)[0]
array-numpy.trunc(array)
我认为2在内存使用方面更快,更便宜…但不确定. python和numpy专家怎么看?
解决方法:
我不是专家,所以我必须使用timeit模块检查速度.我使用IPython(这使计时变得非常容易),但即使没有它,timeit模块也可能是解决之道.
In [21]: a = numpy.random.random((10**6, 3))
In [22]: timeit numpy.modf(a)[0]
10 loops, best of 3: 90.1 ms per loop
In [23]: timeit a-numpy.trunc(a)
10 loops, best of 3: 135 ms per loop
In [24]: timeit numpy.mod(a, 1.0)
10 loops, best of 3: 68.3 ms per loop
In [25]: timeit a % 1.0
10 loops, best of 3: 68.1 ms per loop
最后两个是等效的.我对内存的使用了解不多,但是如果modf(a)[0]和a-numpy.trunc(a)都没有比直接使用mod占用更多的内存,我会感到惊讶.
[顺便说一句,如果您的代码完成了您想要的工作,而您只对改进感兴趣,那么您可能会对the codereview stackexchange感兴趣.我仍然对分隔线的位置没有很好的了解,但是感觉有点多喜欢他们的茶.]
标签:python-2-7,python,numpy
来源: https://codeday.me/bug/20191031/1977642.html