2次元配列の要素を前から3つずつ加算したい

Question

#やりたいこと
numpy.ndarray型の下記のような2次元配列の各インデックス毎の要素を前から3つずつ加算したものを保存したいと思っています。

初学者のため、言葉にするのが難しいため下記に例を挙げます。

index1	index2	index3
1000000	2000000	3000000
100000	200000	300000
10000	20000	30000
1000	2000	3000
100	200	300
10	20	30
1	2	3

print(arr)
# [[1000000, 2000000, 3000000], [100000, 200000, 300000],[10000, 20000, 30000], 
[1000, 2000, 3000], [100, 200, 300], [10, 20, 30], [1, 2, 3]] 

上の例であれば、下記のような値を取りたいです。

# 理想
print(arr1)
# [[1110000, 2220000, 3330000],[1110, 2220, 3330]]

簡単なことだとは思うのですが、上手く頭が回らないためご教授下さい。

Answer 1

サンプルと同じ入出力にしたいとなると、こんな感じでしょうか。

>>> import numpy as np
>>> arr = [[1000000, 2000000, 3000000], [100000, 200000, 300000],[10000,
20000, 30000], [1000, 2000, 3000], [100, 200, 300], [10, 20, 30], [1, 2, 3]]
>>> arr = np.array(arr)
>>> [arr[i:i+3].sum(axis=0).tolist() for i in range(0, arr.shape[0], 3) if arr[i:i+3].shape == (3,3)]
[[1110000, 2220000, 3330000], [1110, 2220, 3330]]

Answer 2

import numpy as np


arr = np.array([[1000000, 2000000, 3000000], [100000, 200000, 300000],[10000, 20000, 30000], 
[1000, 2000, 3000], [100, 200, 300], [10, 20, 30], [1, 2, 3]] )
print(arr)
"""
array([[1000000, 2000000, 3000000],
       [ 100000,  200000,  300000],
       [  10000,   20000,   30000],
       [   1000,    2000,    3000],
       [    100,     200,     300],
       [     10,      20,      30],
       [      1,       2,       3]])
"""
print(arr.shape)
"""
(7, 3)
"""
print(np.sum(arr[0:3], axis=0))
"""
array([1110000, 2220000, 3330000])
"""
print(np.sum(arr[3:6], axis=0))
"""
array([1110, 2220, 3330])
"""

Comments

[wakame]

サンプルの[ 1, 2, 3]は何のために存在しているんだろう。

[fuzzball]

「余りは無視する」という仕様説明のためでしょう。（好意的に解釈するなら）

Answer 3

arr1 = np.sum(arr[:3, :], axis=0)

かな。

スライスで3行目までを切り出して、sumの軸を指定してあげます。試しにaxis=1として結果がどう変わるか確認してみて下さい。