问小矮子。特定索引处的和多维数组
EN

Stack Overflow用户

提问于 2016-11-15 22:40:08

回答 1查看 443关注 0票数 0

我正在尝试替换类似的代码，以便使用numpy进行矢量化的高效操作。

counter = 0
idxs = [1, 3]
lists = [[1, 2, 3, 4], [1, 2, 3, 4], [1, 2, 3, 4]]
for l in lists:
    for idx in idxs:
        counter += l[idx]

回答已采纳

发布于 2016-11-15 22:45:01

只需将数组之和：

idxs = [1, 3]
lists = [[1, 2, 3, 4], [1, 2, 3, 4], [1, 2, 3, 4]]
ary = np.array(lists)
counter = ary[:,idxs].sum()

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/40620999

复制

相似问题

问小矮子。特定索引处的和多维数组EN