python-3.x - 替换列值以生成新的数据框

Question

下面的代码片段:

import pandas as pd

pd.DataFrame(
    {'type': ['A', 'B', 'A', 'C', 'C', 'A'],
     'value': [5, 6, 7, 7, 9, 1]}
)

给出:

  type  value
0    A      5
1    B      6
2    A      7
3    C      7
4    C      9
5    A      1

想要这个:-

pd.DataFrame(
    {'A': [5, 0, 7, 0, 0, 1],
     'B': [0, 6, 0, 0, 0, 0],
     'C': [0, 0, 0, 7, 9, 0]}
)
   A  B  C
0  5  0  0
1  0  6  0
2  7  0  0
3  0  0  7
4  0  0  9
5  1  0  0

我确实尝试使用 for 循环，但努力提高效率。将是一个很大的帮助!

Answer 1

使用 get_dummies 并乘以第二列:

final_df=pd.get_dummies(df['type']).mul(df['value'],axis=0)

   A  B  C
0  5  0  0
1  0  6  0
2  7  0  0
3  0  0  7
4  0  0  9
5  1  0  0