python - pandas根据变量值添加变量

Question

假设以下 pandas 数据框

    Wafer_Id  v1  v2
0          0   9   6
1          0   7   8
2          0   1   5
3          1   6   6
4          1   0   8
5          1   5   0
6          2   8   8
7          2   2   6
8          2   3   5
9          3   5   1
10         3   5   6
11         3   9   8

我想根据 WaferId 对它进行分组，并且我想要得到类似的东西

w
Out[60]: 
   Wafer_Id  v1_1  v1_2  v1_3  v2_1 v2_2 v2_3
0         0     9     7     1     6  ...  ...
1         1     6     0     5     6
2         2     8     2     3     8
3         3     5     5     9     1

我认为我可以使用pivot函数获得结果，但我不知道该怎么做

Answer 1

可能的解决方案

    oes = pd.DataFrame()
    oes['Wafer_Id'] = [0,0,0,1,1,1,2,2,2,3,3,3]
    oes['v1'] = np.random.randint(0, 10, 12)
    oes['v2'] = np.random.randint(0, 10, 12)
    oes['id'] = [0, 1, 2] * 4


    oes.pivot(index='Wafer_Id', columns='id')

oes
Out[74]: 
    Wafer_Id  v1  v2  id
0          0   8   7   0
1          0   3   3   1
2          0   8   0   2
3          1   2   5   0
4          1   4   1   1
5          1   8   8   2
6          2   8   6   0
7          2   4   7   1
8          2   4   3   2
9          3   4   6   0
10         3   9   2   1
11         3   7   1   2

oes.pivot(index='Wafer_Id', columns='id')
Out[75]: 
         v1       v2      
id        0  1  2  0  1  2
Wafer_Id                  
0         8  3  8  7  3  0
1         2  4  8  5  1  8
2         8  4  4  6  7  3
3         4  9  7  6  2  1