Python中的映射函数

Question

文件 score.txt 的内容，列出了玩家在某场比赛中的表现：

80,55,16,26,37,62,49,13,28,56
43,45,47,63,43,65,10,52,30,18
63,71,69,24,54,29,79,83,38,56
46,42,39,14,47,40,72,43,57,47
61,49,65,31,79,62,9,90,65,44
10,28,16,6,61,72,78,55,54,48

以下程序读取文件并将分数存储到列表中

f = open('scores.txt','r')
L = []
for line in f:
    L = L + map(float,str.split(line[:-1],','))
print(L)

但它会导致错误消息。我在课堂上得到的代码非常困惑，对 Pyton 来说非常陌生。 我可以修复代码吗？

Answer 1

看来您已经调整了 python2.x 代码以在 python3.x 中使用。请注意，map 在 python3.x 中不返回列表，它返回一个 generator 映射对象（基本上不是列表），您必须将其转换为list 适当。

此外，我建议使用list.extend 而不是将两者相加。为什么？前者每次加法都会创建一个新的列表对象，比较浪费时间和空间。

numbers = []
for line in f:
    numbers.extend(list(map(float, line.rstrip().split(','))))

print(numbers)

另一种方法是：

for line in f:
    numbers.extend([float(x) for x in line.rstrip().split(',')]) 

这恰好更具可读性。您还可以选择使用嵌套列表理解来摆脱外部for 循环。

numbers = [float(x) for line in f for x in line.rstrip().split(',')]

---

另外，忘记提及这一点（感谢chris in the comments），但您确实应该使用上下文管理器来处理文件 I/O。

with open('scores.txt', 'r') as f:
    ...

它更干净，因为它会在您处理完文件后自动关闭它们。

---

看到您的ValueError 消息后，很明显您的数据存在问题（无效字符等）。让我们尝试一些更激进的东西。

numbers = []
with open('scores.txt', 'r') as f:
    for line in f:
        for x in line.strip().split(','):
            try:
                numbers.append(float(x.strip()))
            except ValueError:
                pass

如果这不起作用，也许，使用正则表达式更激进的东西可能会做到：

import re

numbers = []
with open('scores.txt', 'r') as f:
    for line in f:
        line = re.sub('[^\d\s,.+-]', '', line)
        ... # the rest remains the same