在python中連接二維列表

Question

我正在嘗試使用多個 [1,784] 列表創建一個形狀為 [n,784]（與 MNIST 圖像批次形狀相同）的二維列表。

mylist.append(element)沒有給我我正在尋找的東西，其中mylist是 2d [n,784] 列表， element 是 [1,784] 列表。 它會返回一個形狀為 [n,1,784] 的列表。

我也試過mylist[index].append(element) ，我得到了一個 [784] 1d 列表。

知道如何解決我的問題嗎？

非常感謝

Answer 1

import numpy as np
myarray = np.array(mylist)
newarray = np.concatenate((myarray, element))

如果你想把它變回一個列表：

newlist = newarray.tolist()

Answer 2

a = [[1,1],[2,2]] b = np.concatenate([a, a], axis=1).tolist()

輸出將是：

[[1, 1, 1, 1], [2, 2, 2, 2]]