如何在shape为(1,20,1)的currentStep末尾插入temp并移除首元素？

嘿，我懂你的困扰啦——你想要实现的是移除第二个维度的第一个元素，再把temp追加到该维度的末尾，但之前的代码只是替换了最后一个元素，根本没真正“插入”新元素对吧？这是因为你没有改变数组的长度，只是修改了现有位置的值而已。

正确的实现思路 #

原数组形状是(1,20,1)，我们需要先切掉第二个维度的第一个元素，得到形状(1,19,1)的数组，然后把temp拼接回这个维度的末尾，让数组形状变回(1,20,1)。

具体代码实现 #

这里用NumPy的concatenate函数（推荐，比append更灵活）来完成拼接，步骤如下：

import numpy as np

# 1. 移除第二个维度的第一个元素，得到(1,19,1)的数组
currentStep = currentStep[:, 1:, :]

# 2. 确保temp的形状和拼接维度匹配（必须是(1,1,1)，否则会报错）
# 如果temp是标量/一维数组，先reshape调整形状
temp = temp.reshape(1, 1, 1)

# 3. 沿着第二个维度（axis=1）拼接新元素
currentStep = np.concatenate([currentStep, temp], axis=1)

如果你习惯用np.append，也可以这样写：

currentStep = np.append(currentStep, temp.reshape(1,1,1), axis=1)

为什么之前的代码不对？ #

你写的currentStep[:,-1:,:] = temp只是把现有数组最后一个位置的元素替换成了temp，数组的形状还是(1,19,1)，并没有新增元素。而我们需要的是扩展数组长度，所以必须用拼接的方式来实现。

注意事项 #

一定要保证temp的形状和拼接的维度对齐：比如要沿着axis=1拼接，temp在该维度上的长度必须是1，且其他维度和currentStep一致（也就是(1,1,1)）。如果temp是单个数值，一定要先通过reshape调整形状，否则会出现维度不匹配的错误。

内容的提问来源于stack exchange，提问作者GoBackess