在python中连接2d列表

时间:2018-02-19 04:15:55

标签: python data-structures mnist

我正在尝试使用多个[1,784]列表创建形状为[n,784](与MNIST图像批次形状相同)的2d列表。

mylist.append(element)没有给我我正在寻找的东西,其中mylist是2d [n,784]列表,元素是[1,784]列表。它将返回一个形状为[n,1,784]的列表。

我也试过mylist[index].append(element),而我得到的是[784] 1d列表。

知道如何解决我的问题吗?

非常感谢

2 个答案:

答案 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]]