有没有办法让这段代码变得更好?

时间:2015-01-10 11:03:54

标签: python python-3.x

我的TXT文件包含以下格式的足球结果:

1-0
2-0
2-3
etc...

我想计算每个结果,以了解它们在此文件中的数量。我有代码来计算它:

def find_result(self):
    file = open("results.txt", "r")
    result_00 = 0
    result_01 = 0
    result_02 = 0
    result_03 = 0
    result_10 = 0
    result_11 = 0
    result_12 = 0
    result_13 = 0
    result_20 = 0
    result_21 = 0
    result_22 = 0
    result_23 = 0
    result_30 = 0
    result_31 = 0
    result_32 = 0
    result_33 = 0
    result_other = 0
    results = 0
    for line in file:
        results += 1
        line = line.rstrip()
        if line == '0-0':
            result_00 += 1
        elif line == '0-1':
            result_01 += 1
        elif line == '0-2':
            result_02 += 1
        elif line == '0-3':
            result_03 += 1
        elif line == '1-0':
            result_10 += 1
        elif line == '1-1':
            result_11 += 1
        elif line == '1-2':
            result_12 += 1
        elif line == '1-3':
            result_13 += 1
        elif line == '2-0':
            result_20 += 1
        elif line == '2-1':
            result_21 += 1
        elif line == '2-2':
            result_22 += 1
        elif line == '2-3':
            result_23 += 1
        elif line == '3-0':
            result_30 += 1
        elif line == '3-1':
            result_31 += 1
        elif line == '3-2':
            result_32 += 1
        elif line == '3-3':
            result_33 += 1
        else:
            result_other += 1
    print('[0-0]' + str(result_00))
    print('[0-1]' + str(result_01))
    print('[0-2]' + str(result_02))
    print('[0-3]' + str(result_03))
    print('[1-0]' + str(result_10))
    print('[1-1]' + str(result_11))
    print('[1-2]' + str(result_12))
    print('[1-3]' + str(result_13))
    print('[2-0]' + str(result_20))
    print('[2-1]' + str(result_21))
    print('[2-2]' + str(result_22))
    print('[2-3]' + str(result_23))
    print('[3-0]' + str(result_30))
    print('[3-1]' + str(result_31))
    print('[3-2]' + str(result_32))
    print('[3-3]' + str(result_33))
    print('[OTHER]' + str(result_other))
    print('Matches: ' + str(results))

我认为这不是一个好方法(我的意思是代码太多),有没有更好的解决方案呢?谢谢

2 个答案:

答案 0 :(得分:4)

# import Counter for Counting the number of occurances of each items
from itertools import Counter

# Open the file for reading
with open("results.txt", "r") as input_file:

    # Create the counter object with the lines read from the file
    c = Counter(line.rstrip() for line in input_file)

    # Print the actual item (key) and the number of occurances (count)
    for key, count in c.items():
        print("[{}] - {}".format(key, count))

    # Find the total number of elements encountered, by adding all the counts
    print(sum(c.values())

答案 1 :(得分:3)

作为一种更加pythonic的方式,你可以使用collections.Counter来实现这个目标:

from collections import Counter
c_dic=Counter(open("results.txt", "r").readlines())

并且您有很多选择来打印您的字典或项目,因为您没有嵌套字典,不需要循环项目并打印它们您可以使用json.dumps打印具有任意缩进的字典:

from collections import Counter
import json
print json.dumps(Counter(open("newefile.txt", "r").readlines()),indent=4)