如果我理解正确:
def give_counts(out):
"""Count the fourth element of each cell if it is >0.5
>>> x = [0, 0, 0, 3] # should show up in counts1
>>> o = [0, 0, 0, 0] # should show up in counts2
>>> out = [[x, o, o],
... [x, x, x],
... [o, o, x]]
>>> counts1, counts2 = give_counts(out)
>>> assert counts1 == 5
>>> assert counts2 == 4
"""
values = (True if col[3] > 0.5 else False for row in out for col in row)
counts = collections.Counter(values)
return counts[True], counts[False]
