如何在具有唯一键和序列值的列中唯一求和

Question

这是我的数据框，如下所示：

df = pd.DataFrame({
        'shipment_name':['A','A','A','A','A','B','B','B','B','B','B', 'C','C','C','C','C'],
        'city_name':[2,3,4,5,6,1,2,3,4,5,6,5,6,7,8,9],
        'volume':[46,46,46,46,46,50,50,50,50,50,50,55,55,55,55,55],
        'customer_name':['James','James','James','James','James','James','James','James','James','James','James','James','James','James','James','James']
    })

我想要一个代码，它可以给我一个带有以下值的答案： 城市 2-3 之间的总流量为 (46+50 = 96)，城市 5-6 之间的总流量为 (46+50+55 = 151)。对于名为 James 的客户，他的货运量在 5 和 6 之间为 151，在 2 和 3 之间为 96。

我不明白如何编码。

我不明白如何编码。强文本

Answer 1

请使用SO tour 并阅读有关How to Ask 的信息。这些将帮助您更快地收集有用的回复。

IIUC，此代码应有助于您提出问题：

import pandas as pd
df = pd.DataFrame({
        'shipment_name':['A','A','A','A','A','B','B','B','B','B','B', 'C','C','C','C','C'],
        'city_name':[2,3,4,5,6,1,2,3,4,5,6,5,6,7,8,9],
        'volume':[46,46,46,46,46,50,50,50,50,50,50,55,55,55,55,55],
        'customer_name':['James','James','James','James','James','James','James','James','James','James','James','James','James','James','James','James']
    })
u = df.groupby(['city_name']).sum()

给你：

           volume
city_name        
1              50
2              96
3              96
4              96
5             151
6             151
7              55
8              55
9              55

然后您可以按照this answer 中的步骤创建组合。我在下面做了一些简单的组合：

u = pd.DataFrame(np.abs(np.subtract.outer(u.volume, u.volume)), u.city_name, u.city_name)
stacked = u.stack()
stacked = pd.DataFrame({'cities': stacked.index.to_series(), 'diff': stacked})[['cities', 'diff']].reset_index(drop=True)

请阅读.groupby() 文档here。