如何将 pandas dataframe 转换为张量并重塑它？

Question

我想把一些 dataframe 改成 tensor 让它在 rnn model 上工作。数据的原始形状是这样的。

input : X_train.shape
output : (2101, 32)

预期的形状是

output : (2101, 4, 8)

我试过了

X_train.to_numpy.reshape(-1, 4, 8)

但它没有用。

请帮忙

Answer 1

您正在寻找numpy方法：

X_train.reshape(-1, 4, 8).numpy()