python - 从文件中读取 3d 数组

Question

假设我有一个具有以下值的文件：

1 2
3 4

11 12
13 14

我想把它们读成一个 numpy 2x2x2 数组。标准命令np.loadtxt('testfile')将它们读入大量向量，忽略空格（4x1x8）。我想我可以遍历它们并以正确的方式将它们堆叠在一起，但是我的实际数据文件非常大，如果可能的话，我宁愿不要有太多的 while 循环。在 numpy 系统中是否有一种很好的方法可以做到这一点？

谢谢你的帮助！

Answer 1

使用重塑。

>>> import numpy
>>> a = numpy.loadtxt('testfile')
>>> a
array([[  1.,   2.],
       [  3.,   4.],
       [ 11.,  12.],
       [ 13.,  14.]])
>>> a.reshape((2, 2, 2))
array([[[  1.,   2.],
        [  3.,   4.]],

       [[ 11.,  12.],
        [ 13.,  14.]]])