如何将列添加到根据另一列中的值向上计数的数据框中？

import pandas as pd

# 创建示例数据框
data = {'需要计数的列': ['A', 'B', 'A', 'B', 'A'],
        '计数的列': [0, 0, 0, 0, 0]}
df = pd.DataFrame(data)

# 创建新的空列
df['计数结果'] = 0

# 遍历数据框的每一行
for i in range(len(df)):
    # 获取需要计数的列的值
    value = df.loc[i, '需要计数的列']
    
    # 从当前行向上遍历数据框
    for j in range(i, -1, -1):
        # 找到与当前行相同值的行
        if df.loc[j, '需要计数的列'] == value:
            # 将计数列的值加1
            df.loc[i, '计数结果'] = df.loc[j, '计数结果'] + 1
            break

print(df)