python - 切片 2d numpy 数组

Question

我有一个形状为(2500,500)的numpy数组padded_train_x。

问题是，当我尝试获取该数组元素的形状时，例如padded_train_x[0].shape 它输出 (500,) 但当我作为 padded_train_x[0:1] 运行它时，它输出 (1,500)。为什么会出现这种情况？

我正在尝试使用 keras 在 LSTM 模型中进行预测，但我必须使用 pagged_train_x[0:1] 作为输入，而不是简单地使用 pagged_train_x[0]

Answer 1

这是因为通过 padded_train_x[0:1] 制作切片，你会得到二维数组:

a = np.linspace(1024).reshape(64,-1)
b = a[0]
c = a[0:1]

b
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])
b[0]
0

c
array([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15]])
c[0]
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15])