python - 如何计算 .txt 中的多个输入，而不仅仅是一个

Question

我找到了一个代码，可以计算 .txt 文件中的值列表。但是，由于某种原因，它只会计算出现的第一个值。因此，如果文件具有以下内容：

1
2
2
3
2
4

然后它只会打印：

Final Tally
1:1

The program will automatically shut down in 5 minutes.

完整代码：

    vote = input('Enter your vote: ')
    file = open('votedata.txt', 'a')
    file.write(vote + '\n')
    print('The system is adding your vote. The next person can vote in 3 seconds.')
    time.sleep(3)
    if vote == 'tally':
        break
#end of loop, beginning of tally
from collections import defaultdict
frequencies = defaultdict(int)
for number in open('votedata.txt'):
    frequencies[int(number)] += 1

for number in sorted(frequencies.keys()):
    print(' ')
    print('Final Tally:')
    print(number, ':', frequencies[number])
    print(' ')
    print('The program will automatically shut down in 5 minutes.')
    time.sleep(300)

我如何编写它以便它收集和统计每个整数？

Answer 1

虽然 pandas 可以工作，但我觉得像这样简单的事情有点过分了。

with open("votedata.txt") as fp:
    results = {}
    for row in fp:
        try:
          v = int(float(row))
          if v not in results:
            results[v] = 0
          results[v] += 1
        except:
          print("Invalid Numeric entry")

print( results )

这很简单，不加载额外的库，处理结果，如果用户输入一些垃圾数据，它会在不主动中断程序工作流程的情况下打印错误。

Answer 2

熊猫可以为你做到这一点

df = pd.read_csv('votedata.txt', names=['votes'])
df['votes'].value_counts().to_dict()

{2: 3, 4: 1, 3: 1, 1: 1}

在您的代码中：

import pandas as pd

    vote = input('Enter your vote: ')
    file = open('votedata.txt', 'a')
    file.write(vote + '\n')
    print('The system is adding your vote. The next person can vote in 3 seconds.')
    time.sleep(3)
    if vote == 'tally':
        break

df = pd.read_csv('votedata.txt', names=['votes'])
frequencies = df['votes'].value_counts().to_dict()

for number in sorted(frequencies.keys()):
    print(' ')
    print('Final Tally:')
    print(number, ':', frequencies[number])
    print(' ')
    print('The program will automatically shut down in 5 minutes.')
    time.sleep(300)

Answer 3

您正在遍历所有输出，因此 5 分钟后您将获得第二个结果 :)

您只需要从最后三行中删除缩进即可。