array_shuffle_reshuffle.py

importnumpyasnp


classShuffleMatrix:
'''shuffle matrix '''

defshuffleMatrix(self, matrix, idx: int):
rows=len(matrix)
cols=len(matrix[0])
shuffleMatrix=np.zeros((rows, cols), type(matrix))

# matix shuffling
forrinrange(0, rows):
k=0
forcinrange(0, cols):
shuffleMatrix[r][c] =matrix[r][idx[k]]
k=k+1
returnshuffleMatrix

defreshuffleMatrix(self, matrix, idx: int):
rows=len(matrix)
cols=len(matrix[0])
orignalMatrix=np.zeros((rows, cols), type(matrix))

# matix shuffling
forrinrange(0, rows):
k=0
forcinrange(0, cols):
orignalMatrix[r][idx[k]] =matrix[r][c]
k=k+1
returnorignalMatrix


classShuffleArray:
''' shuffle the values in array and reShuffle the values'''

defshuffle(self, arr, index: int):
'''shuffle the array args {arr:[list or numpy.array], index:[list or numpy.array]}
 index will be the how to shuffle the values according to index values
 '''
lenArray=len(arr)
suffledArr= [] # empty list to store the shuffled list
foriinrange(0, lenArray):
suffledArr.append(arr[index[i]])

returnsuffledArr

defreshuffle(self, shuffledArr, index: int):
'''Reshuffle the array to it's orignal state args = {arr:[list or numpy.array], index:[list or numpy.array]}
 index will be the how to reshuffle the values according through index values'''
lenArray=len(shuffledArr)
orignalArr=np.zeros(lenArray, dtype=type(shuffledArr))
foriinrange(0, lenArray):
orignalArr[index[i]] =shuffledArr[i]

returnorignalArr


lst= [10, 20, 30, 40, 50]
idx= [3, 2, 0, 1, 4]

arrange=ShuffleArray()
lst=arrange.shuffle(lst, idx)
lst=arrange.reshuffle(lst, idx)

matrix= [[2, 5],
 [8, 3]]
arrangeMatrix=ShuffleMatrix()
arrangeMatrix.shuffleMatrix(matrix=matrix, idx=[1, 0])
arrangeMatrix.reshuffleMatrix(matrix=matrix, idx=[1, 0])