所以即时通讯生成了一个以numpy命名为heights的随机列表...

heights = random.randint(10, size=random.randint(20))


我将这个列表用于测试目的,但是我有一个条件,我需要在每次生成时将第一个数字添加到列表中。基本上,我需要为生成的任何随机列表始终创建第一个数字2。
因此,如果列表如下所示:

[1 2 2 5 5 6 7 7 7 9]


我需要它看起来像:

[1 2 2 5 5 6 7 7 7 9 1]


我尝试使用

heights.append(heights[0])


但是,我得到一个错误。
这适用于列表,但不适用于numpy:/

最佳答案

您也可以使用np.concatenate()np.hsatck()

heights = np.concatenate((heights, heights[0:1]))


要么

heights = np.hstack((heights, heights[0:1]))

09-18 08:49