如何在基于目录的数据框中创建动态索引？

Question

我有两个不同的数据框。第一个是不同请求类型的目录以及我目前有多少请求。

data = {'Req Type': ['A', 'B', 'C', 'D'],'Req No': [20, 21, 19, 18]} 
df1 = pd.DataFrame(data) 
print(df1)      

       Req Type  Req No 
0       A        20 
1       B        21 
2       C        19 
3       D        18 

另一个是我这个月有多少新请求的数据框。

data2 = {'Req Type': ['A', 'A', 'C', 'B']}
df2 = pd.DataFrame(data2)
print(df2)
    Req Type  
0       A      
1       A     
2       C     
3       B    

我需要在 df2 中创建一个动态索引，它可以检测每种请求类型的最后一个数字是什么并开始求和。示例：A 的最后一个数字是“20”，因此请求类型“A”的索引是“21”，然后是“22”，对于我的所有请求类型也是如此。

       Req Type  Index 
0       A        21 
1       A        22 
2       C        20 
3       B        22

Answer 1

您可以使用字典并一次性更新它：

import pandas as pd

def get_indices(A, B):
    res, TN = [], dict(zip(A['Req Type'], A['Req No']))
    for t in B['Req Type']:
        TN[t] += 1
        res += [TN[t]]

    B['Index'] = res
    return B


A = pd.DataFrame({'Req Type': ['A', 'B', 'C', 'D'], 'Req No': [20, 21, 19, 18]})
B = pd.DataFrame({'Req Type': ['A', 'A', 'C', 'B']})
print(get_indices(A, B))

打印

  Req Type  Index
0        A     21
1        A     22
2        C     20
3        B     22