Python：获取在 Python 中获得的计数总和

Question

我是一个每天都在与 Python 苦苦挣扎的初学者。我有一个大型数据集，第二列中有动物名称。我有一个程序可以按每个动物的名称来计算每个动物的数量（每行有 1 个动物名称和 1 个“计数”数据）。我正在尝试获取使用 Python 获得的计数数据的总和，但我无法做到这一点。我到目前为止的代码是：

import csv, collections

reader=csv.reader(open('C:\Users\Owl\Data.txt','rb'), delimiter='\t')

counts=collections.Counter()

for line in reader:
   Name=line[1]
   counts[Name]+=1

for (Name, count) in sorted(counts.iteritems()):
   Output=list('%s' % count) #Make output string to a list
   Sum=sum(Output) # Sum function requires a list
   print 'Total kinds of Animals: %s' % Sum

我收到一条错误消息" File "sum_count.py", line 17, in <module> Sum=sum(Output) # Sum function requires a list TypeError: unsupported operand type(s) for +: 'int' and 'str'".

到目前为止，我发现因为sum 显然要求输入类型为列表，所以我将计数数据（字符串）转换为列表，但是当我执行Output=list('%s' % count) 时，似乎所有超过 2 位的计数数据都被拆分。比如我打印Output的时候是这样的：

['1', '6', '3']
['3']
['1', '8', '5', '9']
['7', '9']

而不是

['163']
['3']
['1859']
['79']

我想要在这里做的是获得这些元素的单一“总和”。这里是 4. 四种动物。

我认为这可能是我收到上述错误的原因。我可能错了，但有人可以帮我解决这个问题吗？提前感谢您的帮助！

Answer 1

我认为你不需要使用sum。

试试这个：

for (Name, count) in sorted(counts.iteritems()):
    print 'Species total: %s' % count

或者，可能更好：

for (Name, count) in sorted(counts.iteritems()):
    print 'Total for species %s: %s' % (Name, count)

sum 用于当您有一个数字列表并希望找到该数字列表的总和时。 您已经使用counts 收集了每只动物的总数——您只需要显示它即可。

编辑

要汇总计算的动物总数，您可以这样做：

total = sum(counts.values())
print 'Total number of animals: %d' % total

编辑 2

统计的动物种类数就是counts字典的长度：

print 'Number of kinds of animals: %d' % len(counts)

Answer 2

获取计数

我认为问题源于您将“计数”与“总数”区分开来。 “计数”是该项目出现的总次数。此外，您误用了collections.Counter()，它可以让您的工作变得很多 更轻松。这是我认为您要实现的目标的编码示例：

counts = collections.Counter(line[1] for line in reader if len(line) > 1)
#Now all the occurrences of each item are summed up, AND ordered by number of occurrences

print "Total number of animals: %d" % len(counts)
#This is what I THINK you are trying to do. 

另外：

for name, number in counts.items():
    print "# of %s: %d" % (name, number)

关于你的`TypeError`：

你有一个字符串列表，而不是整数列表。

一个例子：

mylist = ['1', '2', '3']

sum() 所做的只是对可迭代对象执行累积加法，类似于：

total = 0
for item in mylist:
    total = total + item

在这种情况下，total 是 int（值 0），item 是 str（值 '1'）。 Python 不知道如何处理0 + 'string'。

Answer 3

（在评论讨论后重写；原始答案只是指出 OP 试图添加字符串。）

其他答案有更多的扩展机会（所以我会推荐它们），但是如果您只是想快速计算动物类型的数量，您可以简单地计算行数在文件中，并使用您对文件结构的了解。例如，如果您的 csv 文件有一个像 Name, Count, etc. 这样的标题，下一行后面只有您感兴趣的数据，那么动物的数量将是文件中非空行的数量，减去标题。然后，您可以使用以下代码打印计数：

print sum(1 for line in open('test.csv') if line.strip() != '') - 1

以下是该代码的每个部分的作用：

• sum() 在其中添加列表的所有元素。在这种情况下，里面没有一个列表，而是一个生成器表达式，这里可以认为是一个没有进入内存的列表。
• 1 for line in open('test.csv') 这是生成器表达式的第一部分。它本身会生成一个生成器，其长度是test.csv 中的行数，其中每个元素都是1（如果文件中有五行，类似的列表将是[1,1,1,1,1]）。
• if line.strip() != '' 这是生成器表达式的第二部分。它确保只有在行上有任何内容时才会将1 添加到生成器中。
• -1值减一sum(...)返回忽略csv的头部

嗯，我希望在某种程度上有所帮助，而且我应该重申，这种方法只是一种快速而肮脏的方法；例如，如果您正在使用数据做其他事情，您就不会使用它。

Answer 4

首先，您使用的是 Counter 对象，但仅将其用作 defaultdict 的替代品。 如果你想用它来计数，你可以像这样传递你的输入（假设物种出现不止一次，并且你想知道每个物种出现的频率）：

counts = collections.Counter(map(lambda item: item[0], reader))

但是，如果您想计算所有动物的总和（不考虑物种），则必须在第一个循环中添加该计数。正如其他人所说，由于您正在读取字符串，因此您首先必须从该计数中取出一个整数。

import csv

reader=csv.reader(open('in','rb'), delimiter='\t')

counts = dict()

for data in reader:
    animal = data[1]
    if animal not in counts:
        counts[animal]= 1
    else:
        counts[animal]+= 1

for animal in counts:
    print 'Animals of scpecies %s: %s' % (animal, counts[animal])   

print 'Species total: %s' % len(counts)
print 'All animals: %s' % sum(counts.values())