Recherche de moyennes par clé à partir d’un csv en python - python, csv, moyenne

J'ai un simple csv de 2 colonnes et je dois trouver la moyenne par clé c'est à dire. entrée csv

A,2
B,3
A,1
C,2
B,2
D,4
C,2

sortie requise

{"A": 1.5, "B": 2.5, "C": 2, "D": 4}

---

Code jusqu'ici:

pythoncsvfile = open("data.csv")
csv_reader = csv.reader(csvfile, delimiter=",")
for row in csv_reader:
print (row[0],row[1])

Réponses:

2 pour la réponse № 1

Option A

En utilisant csv

import csv
import collections

out = collections.defaultdict(list)
with open("file.csv") as f:
for line in csv.reader(f):
out[line[0]].append(int(line[1]))

for k in out:
out[k] = sum(out[k]) / len(out[k])

print(dict(out))

{"A": 1.5, "B": 2.5, "C": 2.0, "D": 4.0}

---

Option B

En utilisant pandas

import pandas as pd

df = pd.read_csv("file.csv", header=None, names=["Key", "Value"])
out = df.groupby("Key").mean()

print(out.Value.to_dict())

{"A": 1.5, "B": 2.5, "C": 2.0, "D": 4.0}

---

1 pour la réponse № 2

Je pense que vous pourriez utiliser le code suivant:

import csv
from collections import OrderedDict

data = OrderedDict()

with open("data.csv", "rb") as csvfile:
content = csv.reader(csvfile, delimiter=",")
for index, value in content:
if ( not data.has_key(index) ):
#initialize
data[index] = {"times":1, "total":float(value)}
else:
#index already present
data[index] = {"times": data[index]["times"]+1, "total":data[index]["total"]+float(value)}

def average(data):
results = OrderedDict()

for index, values in data.iteritems():
results[index] = values["total"]/values["times"]

return results

print average(data)

OrderedDict([("A", 1.5), ("B", 2.5), ("C", 2.0), ("D", 4.0)])

HTH

---