Python 使用pyton如何将数据集的每一行分配给一个类别，以便每一行都属于一个类别

我不知道如何以植物学友好的方式解释这一点，但我想知道是否可以使用python将数据集中的每一行分配给一个组/类别，但按2列或更多列分层，以便在每个组/类别中混合使用行。我不认为这是一个抽样问题，可能不是groupby，也不确定排序是否会产生一个具有足够随机性的组

Column 1 Column2   Column3
Blue     A         J
Green    A         J
Blue     A         I
Green    A         I
Green    A         I
Blue     B         J
Green    B         I
Blue     B         I
Blue     B         I

会让我

Category Column 1 Column2   Column3
Group 1 Blue     A         J
Group 2 Green    A         J
Group 3 Blue     A         I
Group 1 Green    A         I
Group 3 Green    A         J
Group 1 Blue     B         J
Group 1 Green    B         I
Group 2 Blue     B         I
Group 3 Blue     B         I

有人有什么想法吗？ 谢谢