我有一个看起来像这样的数组:
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)