프로그램

[파이썬] 두 데이터를 같이 shuffle로 섞기

오디세이99 2023. 5. 30. 14:03
728x90
반응형

다음과 같이 x, y 데이터가 있을때 shuffle 적용하기

import numpy as np

x = np.array([1,2,3,4,5])             # 원래 1번쨰 데이터
y = np.array(['a','b','c','d','e'])   # 2번째 데이터

print('x =',x)
x_idx = np.arange(x.shape[0])     # x 의 요소수 만큼 인덱스 값을 만듬
print('x_idx =',x_idx)
np.random.shuffle(x_idx)          # 만들어진 인덱스를 shuffle 함.
print('shuffle x_idx =',x_idx)    
x_new = x[x_idx]                  # x 데이터를 x_idx로 배치
y_new = y[x_idx]                  # y 데이터를 x_idx로 배치
print('shuffle x =',x_new)
print('shuffle y =',y_new)

728x90
반응형