Question

我有一个尺寸为（28,28,60000）的numpy数组，包含60000个28x28图像，表示为像素亮度。我正在尝试对其进行转换，以便我有一个60000 x 784阵列，其中784代表行主要格式的原始28x28图像。我该怎么做呢？我假设我使用numpy.reshape，但我不确定它是如何重新排列的。例如：

[[1,2],        [[1,2,3,4],
 [3,4]]         [5,6,7,8]]
...       -> 
[[5,6],
 [7,8]]

Answer 1

此代码：

import numpy
a = numpy.array([[[1,2],[3,4]],[[5,6],[7,8]]])
print(numpy.reshape(a, (2,4)))

返回：

[[1 2 3 4]
 [5 6 7 8]]

Answer 2

尝试尝试这样的事情：

a = np.arange(6).reshape((3, 2))
b = np.reshape(a, (1, 6))
print a
print b
a = 
[[0 1]
 [2 3]
 [4 5]]
b = [[0 1 2 3 4 5]]

Numpy 3D阵列到2D行主阵列

2 个答案: