迭代列表- Python 3

Question

我有以下代码，用于打开文件，按列对其进行排序，然后将平均值附加到其中。但是，我不能让for循环迭代每一行.x = list(map(int, sortedlist[1:][1][1:]))，我厌倦了把1改为一个名为y的反变量，但是它没有起作用。

下面是文件的样子和代码。

Lee,6,3,4
John,11,10,8
Luke,2,3,8
Terry,4,7,6


import sys, csv, operator
from statistics import mean

#Sorts list
reader = csv.reader(open("O:\\Class 1.csv"), delimiter=",")
sortedlist = sorted(reader, key=operator.itemgetter(1), reverse=True)

#Appends average to the end of each row.
for sublist in sortedlist:
    x = list(map(int, sortedlist[1:][1][1:])) #HERE'S THE PROBLEM!
    x = mean(x)
    sublist.append(x)
    print(sublist) 
print(sortedlist)

Answer 1

您希望得到每个子列表的平均值，所以使用每个子列表而不是排序列表：

for sublist in sortedlist:
    x = mean(map(int, sublist[1:])) #HERE'S THE PROBLEM!
    sublist.append(x)
    print(sublist)
print(sortedlist)

[['Lee', '6', '3', '4', 4.333333333333333], ['Terry', '4', '7', '6', 5.666666666666667], ['Luke', '2', '3', '8', 4.333333333333333], ['John', '11', '10', '8', 9.666666666666666]]

如果您想要组合所有列表的平均值，可以使用itertools.chain从每一行提取所有元素/int并将它们链接在一起：

from itertools import chain

x = mean(map(int, chain.from_iterable(row[1:] for row in sortedlist)))
print(x)
6.0


x = list(map(int, chain.from_iterable(row[1:] for row in sortedlist))
print(x)
[6, 3, 4, 4, 7, 6, 2, 3, 8, 11, 10, 8]