Question

我有一个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] = " "

Answer 1

# 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}

python

1 个答案: