大多数pythonic计算匹配元素的方法是可迭代的

Question

我有一个可迭代的条目,我想收集一些简单的统计数据,比如可以被2整除的所有数字的数量以及可被3整除的所有数字的数量.

我的第一个选择,虽然只迭代列表一次并避免列表扩展(并保持拆分循环重构),看起来相当臃肿:

(alt 1)

r = xrange(1, 10)

twos = 0
threes = 0

for v in r:
  if v % 2 == 0:
    twos+=1
  if v % 3 == 0:
    threes+=1

print twos
print threes

这看起来相当不错,但是有将表达式扩展到列表的缺点:

(alt 2)

r = xrange(1, 10)

print len([1 for v in r if v % 2 == 0])
print len([1 for v in r if v % 3 == 0])

我真正喜欢的是像这样的函数:

(alt 3)

def count(iterable):
  n = 0
  for i in iterable:
    n += 1
  return n

r = xrange(1, 10)

print count(1 for v in r if v % 2 == 0)
print count(1 for v in r if v % 3 == 0)

但这看起来很像没有功能的事情.最后的变体是这样的:

(alt 4)

r = xrange(1, 10)

print sum(1 for v in r if v % 2 == 0)
print sum(1 for v in r if v % 3 == 0)

而最小的(在我的书中可能是最优雅的)它并不觉得它很好地表达了意图.

所以,我的问题是:

您最喜欢哪种方式来收集这些类型的统计数据？如果你有更好的东西,请随意提供自己的选择.

为了澄清下面的一些困惑:

• 实际上,我的过滤谓词比这个简单的测试更复杂.
• 我迭代的对象比数字更大,更复杂
• 我的过滤器函数更加不同,很难参数化为一个谓词

Answer 1

不得不多次迭代列表是不优雅的恕我直言.

我可能会创建一个允许执行以下操作的函数:

twos, threes = countmatching(xrange(1,10),
                             lambda a: a % 2 == 0,
                             lambda a: a % 3 == 0)

一个起点是这样的:

def countmatching(iterable, *predicates):
    v = [0] * len(predicates)
    for e in iterable:
        for i,p in enumerate(predicates):
            if p(e):
                v[i] += 1
    return tuple(v)

顺便说一句,"itertools食谱"有一个很像你的alt4的配方.

def quantify(seq, pred=None):
    "Count how many times the predicate is true in the sequence"
    return sum(imap(pred, seq))

Answer 2

Alt 4!但也许你应该将代码重构为一个函数,该函数接受一个应该包含可分数(两个和三个)的参数.然后你可以有一个更好的功能名称.

def methodName(divNumber, r):
  return sum(1 for v in r if v % divNumber == 0)


print methodName(2, xrange(1, 10))
print methodName(3, xrange(1, 10))