計算CSV文件Python中最后一項的行數是1還是0

Question

我有這樣的CSV文件：

2,1,2,3,1
23,3,2,22,0
2,2,11,2,0
1,2,2,1,1
    .
    .
44,3,3,44,0
2,2,11,2,0

每行以1或0結尾，我想通過計算有最后一項是1或0的行數除以總行數來計算1或0的先驗概率。 如何在python中解決它。 謝謝。

Answer 1

import csv

total = 0
zeros = 0
ones = 0
with open('path/to/file') as infile:
    for row in csv.reader(infile):
        total += 1
        if row[-1] == '0': zeros += 1
        if row[-1] == '1': ones += 1

# do some division to calculate priors