python提取数组元素_python – 获取numpy数组中元素对的总和

77 阅读 0 评论 51 点赞

我是靠谱客的博主风趣战斗机，最近开发中收集的这篇文章主要介绍python提取数组元素_python – 获取numpy数组中元素对的总和，觉得挺不错的，现在分享给大家，希望可以做个参考。

你可以利用NumPy数组的元素加法能力：

In [5]: import numpy as np

In [6]: t = np.array([4, 5, 0, 7, 1, 6, 8, 3, 2, 9])

In [7]: t + np.r_[t[1:],t[0]]

Out[7]: array([ 9, 5, 7, 8, 7, 14, 11, 5, 11, 13])

np.r_是将序列连接在一起以形成新的numpy数组的一种方法.正如我们将在下面看到的,在这种情况下,事实并非如此.

另一种可能性是：

In [10]: t + np.roll(t,-1)

Out[10]: array([ 9, 5, 7, 8, 7, 14, 11, 5, 11, 13])

使用np.roll看起来明显更快：

In [11]: timeit t + np.roll(t,-1)

100000 loops, best of 3: 17.2 us per loop

In [12]: timeit t + np.r_[t[1:],t[0]]

10000 loops, best of 3: 35.5 us per loop

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。