如何在 python 中读取和排序 csv 文件？

Question

我是 python 的新手，有一个带有名称和分数的 csv 文件，如下所示：

Andrew,10
Steve,14
Bob,17
Andrew,11

我需要知道如何读取这个文件，并且该文件必须显示两个同名条目，例如 Andrew,10 和 Andrew,11 为 Andrew,10,11。 我还需要能够按名称、最高分或平均分进行排序。 如果可能，我还希望它只为每个名称使用最后 3 个条目。 这是我试图用来按名称阅读和排序的代码：

with open("Class1.csv", "r") as f:
        Reader = csv.reader(f)
        Data = list(Reader)
        Data.sort()
        print(Data)

Answer 1

Pandas 非常适合它

import pandas as pd

df = pd.read_csv("<pathToFileIN>",index_col=None,header=None)
df.columns = ["name","x"]
n = df.groupby("name").apply(lambda x: ",".join([str(_) for _ in x["x"].values[-3:]])).values
df.drop_duplicates(subset="name",inplace=True)
df["x"] = n
df.sort("name",inplace=True)

df.to_csv("<pathToFileOUT>",index=None,sep=";")

Answer 2

要合并分数，请使用collections.defaultdict ：

scores_by_name = collections.defaultdict(list)
for row in Reader:
    name = row[0]
    score = int(row[1])
    scores_by_name[name].append(score)

要保留最后三个分数，请取 3 个项目切片：

scores_by_name = {name: scores[-3:] for name, score in scores_by_name.items()}

按字母顺序迭代：

for name, scores in sorted(scores_by_name.items()):
    ... # whatever

按最高分迭代：

for name, scores in sorted(scores_by_name.items(), key=(lambda item: max(item[1]))):
    ...