如何根據 Pandas DataFrame 中的條件添加每組具有重復值的新列？

Question

這是一個示例數據幀。

RootProduct | Product | Value
    A           A        1  
    A           B        2   
    A           C        3
    D           D        4
    D           E        5  

當RootProduct == Product按RootProduct分組時，如何添加第四列，重復Value列中的Value ？

這將導致以下 DataFrame

RootProduct | Product | Value  | RootValue
    A           A        1          1
    A           B        2          1
    A           C        3          1 
    D           D        4          4 
    D           E        5          4

Answer 1

想法是通過兩列比較boolean indexing與Series.eq然后創建Series的指數與Product由DataFrame.set_index ，因此可能使用Series.map柱RootProduct ：

s = df[df['RootProduct'].eq(df['Product'])].set_index('Product')['Value']
df['RootValue'] = df['RootProduct'].map(s)
print (df)
  RootProduct Product  Value  RootValue
0           A       A      1          1
1           A       B      2          1
2           A       C      3          1
3           D       D      4          4
4           D       E      5          4

Series詳情：

print (s)
Product
A    1
D    4
Name: Value, dtype: int64