python中两个字典中的值之间的交集

2024-07-23 03:30:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个csv文件,其中包含一些国家的贸易数据。数据格式如下:

rep   par  commodity  value
USA   GER    1         700
USA   GER    2         100
USA   GER    3         400
USA   GER    5         100
USA   GER    80        900
GER   USA    2         300
GER   USA    4         500
GER   USA    5         700
GER   USA    97        450
GER   UK     50        300
UK    USA    4         1100
UK    USA    80        200
UK    GER    50        200
UK    GER    39        650

我打算制作一个新的字典,并通过使用创建的字典,计算国家间共同贸易商品的总价值。 例如,考虑USA-GER之间的贸易,我打算检查GER-USA是否在数据中,如果存在,则对常见商品的价值进行求和,并对所有国家执行相同的操作。字典应该是这样的:

Dic_c1c2_producs=
{('USA','GER'): ('1','700'),('2','100'),('3','400'),('5','100'),('80','900'); 
('GER','USA'):('2','300'),('4','500'),('5','700'),('97','450') ;
('GER','UK'):('50','300'); 
('UK','USA'): ('4','80'),('80','200'); 
('UK','GER'): ('50','200'),('39','650')}

如你所见,USA-GER和GER-USA有共同的商品2和商品5,这些商品的价值是(100+300)+(100+700)。 对于美国-英国和英国-美国,我们有共同的商品:0,所以总贸易额也将为0。对于GER-UK和UK-GER,商品50是共同的,贸易总额是300+200。 最后,我想要的是:

Dic_c1c2_summation={('USA','GER'):1200;('GER','UK'):500; ('UK','USA'):0}

任何帮助都将不胜感激。你知道吗

除了我的帖子,我还写了以下几行:

from collections import defaultdict
rfile = csv.reader(open("filepath",'r'))
rfile.next()
dic_c1c2_products = defaultdict(set) 
dic_c_products = {}
country = set()
for row in rfile :
      c1 = row[0]
      c2 = row[1]
      p = row[2]
      country.add(c1)
for i in country :
    dic_c_products[i] = set()
rfile = csv.reader(open("filepath"))
rfile.next()
for i in rfile:
    c1 = i[0]
    c2 = i[1]
    p = i[2]
    v=i[3]
    dic_c_products[c1].add((p,v))
    if not dic_c1c2_products.has_key((c1,c2)) :
        dic_c1c2_products[(c1,c2)] = set()
        dic_c1c2_products[(c1,c2)].add((p,v))

    else:
            dic_c1c2_products[(c1,c2)].add((p,v))
c_list = dic_c_products.keys()
dic_c1c2_productsummation = set()
for i in dic_c1c2_products.keys():
    if dic_c1c2_products.has_key((i[1],i[0])):
       for p1, v1 in dic_c1c2_products[(i[0],i[1])]:
           for p2, v2 in dic_c1c2_products[(i[1],i[0])]:
               if p1==p2:
                  summation=v1+v2
                  if i not in dic_c1c2_productsum.keys():
                       dic_c1c2_productsum[(i[0],i[1])]=(p1, summation)
                  else:
                       dic_c1c2_productsum[(i[0],i[1])].add((p1, summation))
    else:
          dic_c1c2_productsn[i] = " "

Tags: inaddfor商品productsc2setuk
1条回答
网友
1楼 · 发布于 2024-07-23 03:30:00
# save your data in a file called data
import pandas as pd
data = pd.read_csv('data', delim_whitespace=True)
data['par_rep'] = data.apply(lambda x: '_'.join(sorted([x['par'], x['rep']])), axis=1)
result = data.groupby(('par_rep', 'commodity')).filter(lambda x: len(x) >= 2).groupby(('par_rep'))['value'].sum().to_dict()

最后result{'GER_UK': 500, 'GER_USA': 1200}

相关问题 更多 >