ndarrayで内包表記で二つの配列を計算

a = [[-421553.62791753 -116705.19767956 -114233.25168793]
[-464553.46847843 -145207.21261045 -122150.05355102]]
b = [1.06476717e+35 4.24943653e+34]
という二つの配列があります。

np.shape(a) = (2,3)
np.shape(b) = (2,)
となっております。

このとき、
c = [a[0,:]/b[0]
a[1,:]/b[1]]
となって、
np.shape(c) = (2,3)
となるような計算をしたいです。将来的にはこれを拡張してnp.shape(a) = (N,3), np.shape(b) = (N,)の配列で行いたく思います。

for文でcを求めることはできそうですが、内包表記等でcを生成することは可能なのでしょうか?

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

回答2件

b を転置(transpose)する方法もあります。

python
1
2print(a / b[:,None])
3
4#
5array([[-3.95911557e-30, -1.09606307e-30, -1.07284724e-30],
6       [-1.09321192e-29, -3.41709334e-30, -2.87450001e-30]])

投稿2021/12/06 11:48

melian

総合スコア20655

Lily_1007

2021/12/06 12:35

b.Tの転置は上手くいかなかったので，1次元配列では[:, None]のように転置ができるのですね！参考になりました。ありがとうございます！

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

ベストアンサー

aとｂがnumpy.ndarrayなら、ｂのshapeを変えて割り算するだけです。

python
1>>> print(a)
2[[-421553.62791753 -116705.19767956 -114233.25168793]
3 [-464553.46847843 -145207.21261045 -122150.05355102]]
4>>> print(b)
5[1.06476717e+35 4.24943653e+34]
6>>> print(a/b.reshape(2,-1))
7[[-3.95911557e-30 -1.09606307e-30 -1.07284724e-30]
8 [-1.09321192e-29 -3.41709334e-30 -2.87450001e-30]]