特定の要素に対する和を取り出し、新たに配列を生成する方法

前提・実現したいこと

・下記のコードのようなxに対するyの和を取ったものを配列(またはリスト)で表したいと考えているのですが、どのようにすればよろしいのか教えていただきたいです。

ソースコード

Python3
1import numpy as np
2x = np.array([1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5])
3y = np.array([0, 0, 0, 0, 3, 3, 3, 0, 0, 0, 0, 3, 3, 3, 3, 3, 0, 0])
4
5xy = np.c_[x, y]
6xy
7
8array([[1, 0],
9       [1, 0],
10       [1, 0],
11       [1, 0],
12       [2, 3],
13       [2, 3],
14       [2, 3],
15       [3, 0],
16       [3, 0],
17       [3, 0],
18       [3, 0],
19       [4, 3],
20       [4, 3],
21       [4, 3],
22       [4, 3],
23       [4, 3],
24       [5, 0],
25       [5, 0]])

(おおよそ)実現したい結果

Python3
1array([[1, 0],
2       [2, 9],
3       [3, 0],
4       [4, 15],
5       [5, 0]])

行動規範の内容に同意します

回答1件

ベストアンサー

numpy 使えば簡単にできますね。

方法1

import numpy as np

x = np.array([1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5])
y = np.array([0, 0, 0, 0, 3, 3, 3, 0, 0, 0, 0, 3, 3, 3, 3, 3, 0, 0])

sums = []
for label in  np.unique(x):
    sums.append(y[x == label].sum())
print(sums)  # [0, 9, 0, 15, 0]

方法2

np.bincount()

python
1import numpy as np
2
3x = np.array([1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5])
4y = np.array([0, 0, 0, 0, 3, 3, 3, 0, 0, 0, 0, 3, 3, 3, 3, 3, 0, 0])
5
6sums = np.bincount(x, weights=y)[1:]
7print(sums)  # [ 0.  9.  0. 15.  0.]