小编Ano*_*nom的帖子

numpy数组:按一列分组,求和另一列

我有一个看起来像这样的数组:

 array([[ 0,  1,  2],
        [ 1,  1,  6],
        [ 2,  2, 10],
        [ 3,  2, 14]])
Run Code Online (Sandbox Code Playgroud)

我想对第二列中具有相同值的第三列的值求和,所以结果是:

 array([[ 0,  1,  8],
        [ 1,  2, 24]])
Run Code Online (Sandbox Code Playgroud)

我开始对此进行编码,但是我坚持下面的总和:

import numpy as np
import sys

inFile = sys.argv[1]

with open(inFile, 'r') as t:
    f = np.genfromtxt(t, delimiter=None, names =["1","2","3"])

f.sort(order=["1","2"])
if value == previous.value:
   sum(f["3"])
Run Code Online (Sandbox Code Playgroud)

python arrays numpy

6
推荐指数
3
解决办法
3002
查看次数

标签 统计

arrays ×1

numpy ×1

python ×1