3

私は持っている：

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

数値とその下の数値の合計である配列を取得するにはどうすればよいですか:

array([[7,  9,  11,],   # 1+6=7 , 2+7=9 , 3+8=11
       [17, 19, 21,],
       [27, 29, 31,]])

元の配列を反復してこれを行いたいと思います。何かのようなもの：

b[x,y] = [a[x,y]+a[x+1,y] for x in range(0,3) for y in range(1,4)] #ERROR!

ただし、これは構文上の理由で機能しません。誰かが私に適切な構文を教えてもらえますか? 私はプロのプログラマーではなく、Python は初めてです。前もって感謝します。

4

1 に答える 1

7

スライスを使用して、(1) 配列の最後の行を除くすべて、および (2) 配列の最初の行を除くすべてを選択します。次に、それらを追加します。

>>> a[:-1,:] + a[1:,:]
array([[ 5,  7,  9, 11, 13],
       [15, 17, 19, 21, 23],
       [25, 27, 29, 31, 33],
       [35, 37, 39, 41, 43]])
于 2013-11-11T03:37:57.120 に答える