每次在Python中運行代碼時，如何以相同的順序隨機化pandas列？

Question

這是我的代碼：

random_idx = np.random.permutation(len(cIds))
train_Ids = cIds[random_idx[:train_size]]

現在，我希望每次運行這行代碼時，列表都以相同的順序隨機化。

注意：我不想在文本文件中保存random_idx變量，並獲取相同的列表。

Answer 1

您可以使用seed來告訴numpy生成相同的隨機數：

np.random.seed(seed=1234)
random_idx = np.random.permutation(len(cIds))

如同：

np.random.seed(1234)
random_idx = np.random.permutation(len(cIds))

要么

random_idx = np.random.RandomState(seed=1234).permutation(len(cIds)

種子：必須可以轉換為32位無符號整數

Answer 2

或者你可以用熊貓風格做：

cIds.sample(n=train_size, 
            replace=False, 
            random_state=your_favorite_number_here)