我正在尝试使用多个[1,784]列表创建形状为[n,784](与MNIST图像批次形状相同)的2d列表。
mylist.append(element)
没有给我我正在寻找的东西,其中mylist
是2d [n,784]列表,元素是[1,784]列表。它将返回一个形状为[n,1,784]的列表。
我也试过mylist[index].append(element)
,而我得到的是[784] 1d列表。
知道如何解决我的问题吗?
非常感谢
答案 0 :(得分:0)
import numpy as np
myarray = np.array(mylist)
newarray = np.concatenate((myarray, element))
如果你想把它变成一个列表:
newlist = newarray.tolist()
答案 1 :(得分:0)
a = [[1,1],[2,2]]
b = np.concatenate([a, a], axis=1).tolist()
输出将是:
[[1, 1, 1, 1], [2, 2, 2, 2]]