从2D矩阵生成器创建3D矩阵

Question

我正在使用pandas读取许多.csv文件。 每个文件将产生一个3x5数据帧。

dtx = (pd.read_csv(f).values for f in get_filelist(datadirectory))

我想合并来自各个文件的所有数据，以创建一个3维数组（如果我有10个文件，那么我想得到一个形状为10x3x5的数组）

我可以创建一个空的python列表，并使用for循环附加在dtx找到的所有数组，但是我想要一个更多的pythonic解决方案。 我努力了

np.concatenate([tf for tf in dtx])

没有理想的效果。 如何将从.csv文件读取的所有数据连接到一个大型3D阵列中？

Answer 1

您可以向数组添加新维度并将其串联：

dtx = (pd.read_csv(f).values for f in get_filelist(datadirectory))
np.concatenate([tf[np.newaxis] for tf in dtx], axis=0)

例：

np.concatenate([tf[np.newaxis] for tf in (np.arange(4).reshape((2,2)) for i in range(3))], axis=0)
>> array([[[0, 1],
        [2, 3]],

       [[0, 1],
        [2, 3]],

       [[0, 1],
        [2, 3]]])