python - 一维数组到二维数组的 Numpy 列表

Question

我有一个包含通过 numpy.save 腌制的 2D numpy 数组的大型列表文件。我正在尝试读取每个文件的第一列并创建一个新的二维数组。

我目前使用 numpy.load 和 mmap 读取每一列。一维数组现在在一个列表中。

col_list = []
for f in file_list:
    Temp = np.load(f,mmap_mode='r')
    col_list.append(Temp[:,0])

如何将其转换为二维数组？

Answer 1

你可以使用

numpy.stack(arrays, axis=0)

如果你有一个数组数组。如果要堆叠列而不是行，可以指定轴。