pythonでリストを一つにまとめたい

Question

初歩的な質問ですが、pythonで機械学習をしていて、クラス分類結果のバイナリリストを作成したいのですが、リストがバッチ数32個ごとに格納されているためうまくいきません。

これを↓

[array([1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 0,
       1, 0, 1, 1, 0, 0, 1, 1, 1, 1]), array([1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 0, 1,
       0, 1, 0, 0, 1, 1, 0, 1, 1, 0]), array([1, 0, 1])]

こうしたい↓

[1, 0, 0, 1, 1, 0, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 0,
       1, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1, 1, 0, 1, 0, 1]

どなたか方法を教えていただけると幸いです。

追記
pytorchで取得したラベルの予測結果をバイナリリストに格納したいです。

_, preds = torch.max(outputs, 1)  # ラベルを予測
     
preds_all.append(preds.cpu().numpy())

Comments

[8524ba23]

そのような結果になるソースコードを提示ください。

Answer 1

元のpreds_allがどんな変数が不明ですが、以下のようなコードであればほしい結果が得られるかと思います。

import numpy as np

preds_all = np.array([],dtype=int)
for a in [np.array([0,1]), np.array([1,0])]:
    preds_all = np.append(preds_all, a)

print(preds_all) # [0 1 1 0]

Comments

[kotamu]

for文のinの中にくっつけたい配列を書き連ねるという解釈でよいのでしょうか？実際は32×300くらいの配列数を扱うことになるので、手動で書くとなると少し厳しいです。。

[8524ba23]

くっつけたい配列がどのようなものかは分かりませんが、一般的には
「for a in くっつけたい配列:」でよいかと思います。

[kotamu]

すみません。質問の内容が不十分でした。preds_allというリストに上記のarray[]が三つ格納されていて、3×32の形になっています。これを1×32の形にしたいというのが今回の目的になります。

[kotamu]

解決しました。お手数おかけしました。

Answer 2

appendを使えばいいと思います

Comments

[kotamu]

list = []
for i in range(3):
for j in range(32):
list.append(array_all[i][j])
print(list)
とすると三つ目の配列がうまく入らずエラーとなります。
具体的に教えていただけると助かります。