我有这样的行的文件
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)
我非常坚持。首先,我从这个生成字典有不正确的总数,也是我根本不知道如何按值排序的字典
这是使用csv
和collections
模块的一个方法
例如:
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)])
使用熊猫:
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]]