随机地以相同的顺序混合来自不同文件的数据和标签

时间:2017-04-05 10:50:29

标签: python numpy random shuffle

我有两个numpy数组,第一个包含数据,第二个包含标签。 我想根据标签调整数据。换句话说,如何以相同的顺序对标签和数据进行随机播放。

import numpy as np
data=np.genfromtxt("dataset.csv", delimiter=',')
classes=np.genfromtxt("labels.csv",dtype=np.str , delimiter='\t')

x=np.random.shuffle(data)
y=x[classes]

这样可以保留改组的顺序吗?

2 个答案:

答案 0 :(得分:12)

使用np.random.permutation生成元素的随机顺序,并简单地将数组索引到dataclasses -

idx = np.random.permutation(len(data))
x,y = data[idx], classes[idx]

答案 1 :(得分:0)

或者,您可以将数据和标签连接在一起,将它们混洗,然后将它们分为输入x 标签y ,如下所示:

def read_data(filename, delimiter, datatype): # Read data from a file
    return = np.genfromtxt(filename, delimiter, dtype= datatype)

classes = read_data('labels.csv', dtype= np.str , delimiter='\t')
data = read_data('data.csv', delimiter=',')
dataset = np.r_['1', data, classes] # Concatenate along second axis

def dataset_shuffle(dataset): # Returns separated shuffled data and classes from dataset 
    np.random.shuffle(dataset)
    n, m = dataset.shape
    x = data[:, 0:m-1]
    y = data[:, m-1]
    return x, y # Return shuffled x and y with preserved order