我有一个文本文档,格式为:
-1+1
-1-1
+1+1
-1-1
+1-1
...
我想要一个程序来计算有多少行是-1+1行和+1-1行。然后,程序只需要返回像这样的行数的值。
我已经写好了代码:
f1 = open("results.txt", "r")
fileOne = f1.readlines()
f1.close()
x = 0
for i in fileOne:
if i == '-1+1':
x += 1
elif i == '+1-1':
x += 1
else:
continue
print x
但是由于某种原因,它总是返回0,我不知道为什么。
用collections.Counter
代替:
import collections
with open('results.txt') as infile:
counts = collections.Counter(l.strip() for l in infile)
for line, count in counts.most_common():
print line, count
最重要的是,在计算行数时删除空白(特别是换行符,但任何其他空格或制表符也可能会干扰)。
.readlines()
在行中留下n
,这就是它们不匹配的原因。
如果您不想导入模块,请使用简短的代码,并喜欢一点'list'的理解:
with open('results.txt') as infile:
counts = { key: infile.count(key) for key in ['-1+1', '+1-1'] }
然后访问counts
作为字典