numpy.array内のリストの処理がおかしい

Question

numpy.array内のリストの処理がおかしい

具体的でないタイトルですみません。
numpy arrayでの処理で、
array([[1,2,3,4]])
のような出力を得たいのですが、どのようにしても
array(list([1,2,3,4]))
のような出力を得てしまいます。

該当のソースコード

vector = [[1,2,3,4][5,6,7,8]]
max_size = 6

vec_list = []
for i in range(len(vector)):
    while len(vector[i]) < max_size:
         # 末尾に0を追加       
        vector[i].append(0)
    
    vec_list.append(vector[i])
vec_array = np.array(vec_list)

得たいアウトプット・実際のアウトプット

# 得たいアウトプット
vec_array = array([[1,2,3,4,0,0],[5,6,7,8,0,0]])

# 実際のアウトプット
vec_array = array([list([1,2,3,4,0,0]),list([5,6,7,8,0,0])])

こちらはKerasの分類予測モデルに入力するために処理しているのですが、上の出力ではmodel.predictが可能な一方、下の出力ではshapeが違うと怒られてしまいます...
上の出力と下の出力では一体何が違うのか、なぜこのように出力されるのか、どんなアドバイスでも構いませんので、ご教授いただきますと幸いです。
どうぞよろしくお願いいたします。

Answer 1

質問文のコード自体は（数カ所修正が要りますが）問題なく動くようです。

import numpy as np
vector = [[1,2,3,4],[5,6,7,8]]
max_size = 6

vec_list = []
for i in range(len(vector)):
    while len(vector[i]) < max_size:
         # 末尾に0を追加       
        vector[i].append(0)

    vec_list.append(vector[i])
vec_array = np.array(vec_list)
print(repr(vec_array))
"""
array([[1, 2, 3, 4, 0, 0],
       [5, 6, 7, 8, 0, 0]])
"""

想像ですが、実際のコードでは子リストの長さが異なるのではないでしょうか。

>>> import numpy as np
>>> np.array([[1,2,3],[4,5,6,7]])
array([list([1, 2, 3]), list([4, 5, 6, 7])], dtype=object)

いわゆる「四角い」データでないとnumpy配列に落とし込むことはできません。

Comments

[glass]

おっしゃる通りでした！本当にありがとうございました！！