Numpy 3次元配列作成

Question

aの形状、bの値で配列を作成したいと思っています。(a、ｂの1次元の要素数は同じになります)

np.zeros_likeのような処理を期待しているのですが、どのように実装すればよいかご教授いただけないでしょうか。

よろしくお願い致します。

a =[[[0 0 0]
     [1 1 1]],
    [[2 2 2]
     [3 3 3]],
    [[4 4 4]
     [5 5 5]]]

b = [10, 20, 30]

期待値 = [[[10 20 30]
           [10 20 30]],
          [[10 20 30]
           [10 20 30]],
          [[10 20 30]
           [10 20 30]]]

Answer 1

import numpy as np

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

b = np.array([10, 20, 30])

c = np.ones(a.shape)*b
print(c)
""" =>
[[[10. 20. 30.]
  [10. 20. 30.]]

 [[10. 20. 30.]
  [10. 20. 30.]]

 [[10. 20. 30.]
  [10. 20. 30.]]]
"""

Comments

[daodao]

回答ありがとうございます！
期待通りの結果になりました。