对 NumPy 数组中连续值的切片求和
- 作者: 带着媳妇儿奔小康
- 来源: 51数据库
- 2022-10-21
问题描述
假设我有一个包含 10 个值的 numpy 数组 a.这里只是一个示例情况,尽管我想对长度为 100 的数组重复相同的操作.
Let's say I have a numpy array a containing 10 values. Just an example situation here, although I would like to repeat the same for an array with length 100.
a = np.array([1,2,3,4,5,6,7,8,9,10])
我想将前 5 个值与后 5 个值相加,以此类推,并将它们存储在一个新的空列表中,例如 b.
I would like to sum the first 5 values followed by the second 5 values and so on and store them in a new empty list say b.
所以 b 将包含 b = [15,40].
我该怎么做呢?
推荐答案
试试这个列表推导:
b = [sum(a[current: current+5]) for current in xrange(0, len(a), 5)]
它一次从列表中取出 5 个切片,将它们相加并构造一个列表.也适用于长度不是 5 的倍数的列表.
It takes slices of 5 at a time from the list, sums them up and constructs a list. Also works for lists which aren't a multiple of 5 in length.
(xrange在python3+中应该是range)
(xrange should be range in python3+)
推荐阅读
热点文章
Discord.py(重写)on_member_update 无法正常工作
0
Discord.py 在 vc 中获取用户分钟数
0
discord.py 重写 |为我的命令出错
0
Discord.py rewrite 如何 DM 命令?
0
播放音频时,最后一部分被切断.如何解决这个问题?(discord.py)
0
在消息删除消息 Discord.py
0
如何使 discord.py 机器人私人/直接消息不是作者的人?
0
(Discord.py) 如何获取整个嵌入内容?
0
Discord bot 尽管获得了许可,但不能提及所有人
0
Discord.py discord.NotFound 异常
0
