按组洗牌大熊猫据帧

Question

我的数据框看起来像这样

sampleID col1 col2 
    1  1 63 
    1  2 23 
    1  3 73 
    2  1 20 
    2  2 94 
    2  3 99 
    3  1 73 
    3  2 56 
    3  3 34 

我需要洗牌数据框保持相同的样品一起和COL1的顺序必须相同上述数据帧。

所以我需要像这样

sampleID col1 col2 
    2  1 20 
    2  2 94 
    2  3 99 
    3  1 73 
    3  2 56 
    3  3 34 
    1  1 63 
    1  2 23 
    1  3 73 

我怎样才能做到这一点？如果我的例子不清楚，请让我知道。

Answer 1

假设你想要洗牌sampleID。首先df.groupby，洗牌（import random第一），然后调用pd.concat：

In [423]: groups = [df for _, df in df.groupby('sampleID')] 

In [424]: random.shuffle(groups) 

In [427]: pd.concat(groups).reset_index(drop=True) 
Out[427]: 
    sampleID col1 col2 
0   2  1 20 
1   2  2 94 
2   2  3 99 
3   1  1 63 
4   1  2 23 
5   1  3 73 
6   3  1 73 
7   3  2 56 
8   3  3 34 

您与df.reset_index(drop=True)重置索引，但它是一个可选的步骤。