从分组值仅使用标准库的另一列由排序累计一列的一个文本文件?

问题描述 投票:1回答:2

我有这样的行的文件

id, car_type, cost
1, benz, 60000
2, benz, 55000
3, bmw, 30000
4, benz, 25000
5, bmw, 26000
6, ford, 5000

我想排序每个car_type总成本这个文件。总成本,例如,对于“benz”将60000 + 55000 + 25000 = 14000

所以最终的输出应

benz, 140000
bmw, 56000
ford, 5000

到目前为止,这是我有:

file = "small_sample.txt"


f=open(file,"r")
lines=f.readlines()[1:]
car_and_cost ={}
for x in lines:
    cost = x.split(',')[4].rstrip('\n')
    car_and_cost.update({x.split(',')[3]:float(cost)})
f.close()
print(car_and_cost)

new_dic = {}
for key,lis in car_and_cost.items():
    new_dic[key] = sum(lis)
print(new_dic)

我非常坚持。首先,我从这个生成字典有不正确的总数,也是我根本不知道如何按值排序的字典

python sorting sum cumulative-sum accumulate
2个回答
3
投票

这是使用csvcollections模块的一个方法

例如:

import csv
from collections import defaultdict, OrderedDict

result = defaultdict(int)

with open(filename) as infile:
    reader = csv.DictReader(infile)
    for row in reader:                    #Iterate Each row
        result[row[" car_type"]] += int(row[" cost"])   #Add costs

print(OrderedDict(sorted(result.items(), key=lambda x: x[1], reverse=True)))

输出:

OrderedDict([(' benz', 140000), (' bmw', 56000), (' ford', 5000)])

1
投票

使用熊猫:

import pandas as pd
df = pd.read_csv(logFile)

result = df.groupby(' car_type').sum()
print(result)

OUTPUT:

           id    cost
 car_type            
 benz       7  140000
 bmw        8   56000
 ford       6    5000

编辑:

logFile = "tem.csv"
array = []
import csv

with open("tem.csv", "r+") as fin:
    for row in csv.reader(fin):
        array.append(row[1:])

dd = {k: 0 for k in dict(array).keys()}
for x in array: dd[x[0]] += int(x[1])
print(dd)

OUTPUT:

{' benz': 140000, ' bmw': 56000, ' ford': 5000}

或者如果你想他们的列表:

print([[k,v] for k,v in  dd.items()])

OUTPUT:

[[' benz', 140000], [' bmw', 56000], [' ford', 5000]]
© www.soinside.com 2019 - 2024. All rights reserved.