python-实用操作-同序shuffle-按相同顺序打乱两个数组

最新推荐文章于 2024-02-02 09:15:28 发布

shangyj17

最新推荐文章于 2024-02-02 09:15:28 发布

阅读量1.6w

点赞数 56

CC 4.0 BY-SA版权

分类专栏： python 基础

本文链接：https://blog.youkuaiyun.com/qq_17753903/article/details/82181325

python 同时被 2 个专栏收录

82 篇文章

订阅专栏

基础

10 篇文章

订阅专栏

在网络训练中，随机打乱训练数据可获更好效果，且images和labels需按相同顺序打乱以保证对应关系。可考虑使用numpy.random.get_state()和numpy.random.set_state()函数，通过保存和重新载入状态，能在保证对应关系不变的情况下完成随机打乱。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

应用场景：

在网络训练过程中，将训练数据随机打乱能够过的更好的训练效果；

训练的images 和 labels 需要按照相同的顺序打乱，以保证image--label的对应关系；

这时候可以考虑numpy.random.get_state() , numpy.random.set_state() 这组函数；

两个函数说明如下：

https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.get_state.html#numpy.random.get_state

https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.set_state.html#numpy.random.set_state

示例：

import numpy as np

a = np.arange(0,10,1)
b = np.arange(10,20,1)
print(a,b)
#result:[0 1 2 3 4 5 6 7 8 9] [10 11 12 13 14 15 16 17 18 19]
state = np.random.get_state()
np.random.shuffle(a)
print(a)
#result:[6 4 5 3 7 2 0 1 8 9]
np.random.set_state(state)
np.random.shuffle(b)
print(b)
#result:[16 14 15 13 17 12 10 11 18 19]

通过get_state()保存状态，set_state()重新载入状态，可以使得 a,b能够在保证对应关系不变的情况下，完成随机打乱。