ベクトルから自身の転置と掛けた行列を生成したい

Question

以下のコードのように、あるベクトルを自身の転置と掛けて行列を生成したいのですが、どうやればいいのかわからず困っています。

import numpy as np

vec = np.array(range(6))
np.reshape(vec, (6,1))
print(vec.shape) #ここで(6,1)になってほしい

mat = np.dot(vec, vec.T)
print(mat) #6×6行列が欲しい

実行結果
(6,)
55

初歩的なことで恐縮なのですが、軽くググってわからなかったので質問させていただきます。

Answer 1

以下です。

>>> import numpy as np
>>>
>>> vec = np.array(range(6)).reshape(6,1)
>>> mat = np.dot(vec, vec.T)
>>> print(mat)
[[ 0  0  0  0  0  0]
 [ 0  1  2  3  4  5]
 [ 0  2  4  6  8 10]
 [ 0  3  6  9 12 15]
 [ 0  4  8 12 16 20]
 [ 0  5 10 15 20 25]]

Comments

[s.s.]

reshapeって変換結果返り値で受け取るんですね……
ありがとうございました。

Answer 2

reshapeした結果を代入してください。
np.reshape(vec, (6,1))→vec = np.reshape(vec, (6,1))