python - 使用 numpy 导入数组文件

Question

我有一个包含多个 .npy 文件(numpy 数组)的目录，每个文件都有一个二维数组(相同的宽度和高度)。我需要读取所有文件并生成一个包含目录中所有数组的 3 维数组，结果形状应该类似于(文件数、宽度、高度)。

到目前为止我的代码:

import os
import numpy

for file in os.listdir(os.getcwd()):
   result = numpy.load(file) #Obviously this doen't work

但我就是不知道如何生成结果数组。我应该先创建一个零数组然后填充它吗？我可以即时进行吗？你能帮帮我吗？

Answer 1

如果您知道有多少个以及大小是多少，请先创建一个空数组。 (空数组更快，因为您不必将所有元素归零。)像这样:

# Allocate empty array.
bigarray = numpy.empty([width, height, len(filenames)]);
# Load files.
for i in range(len(filenames)):
    bigarray[:,:,i] = numpy.load(filenames[i]);

如果您事先不知道尺寸，请使用 numpy.append。这相当慢，因为它必须在每次迭代中分配新的内存块并复制数据。试试这个:

# Load first array.
bigarray = numpy.load(filenames[0]);
# Add a new axis to make it 3D.
bigarray = bigarray[numpy.newaxis,...];
# Load rest of arrays.
for i in range(1,len(filenames)):
    bigarray = numpy.append(bigarray, numpy.load(filenames[i])[numpy.newaxis,...], axis=0);