numpy行列の(5, 1)にreshapeする方法

Question

numpyの次の文字列をshape=(5, 1)に変換しようとすると、shape=(1,5)になってしまします。

# [image_path, x1, y1, x2, y2, class_id]
df = ['2007_000027.jpg', '174', '101', '349', '351', '14',
     '2007_000027.jpg', '174', '101', '349', '351', '14']
for d in df:
    d=d.strip().split(',')
    box = np.array([np.array(list(map(int,box.split(',')))) for box in d[1:]])
  
Box 
>>>
[[174],
[101],
[349],
[351],
[14]] 

それを次のように

[[174],[101], [349], [351], [14]]

reshapeを使用せず、shape=(5, 1)の形にするにはどうしたら良いでしょうか？

Answer 1

内包記法で以下のように作成することもできます。

df = [
    "2007_000027.jpg",
    "174",
    "101",
    "349",
    "351",
    "14",
]

box = np.array([[x] for x in df[1:]])
print(box)
# [['174']
#  ['101']
#  ['349']
#  ['351']
#  ['14']]

Comments

[trafalbad]

reshapeを使用せずにshape=(5, 1)の形にしたいのですがどうしたら良いでしょうか？

[tiitoi]

回答のコードは reshape() を使用しないで、shape が (5, 1) とできていますが、なにか問題がありますか？