python: lấy số lượng mục từ danh sách (chuỗi) với điều kiện nhất định

Giả sử rằng tôi có một danh sách với số lượng lớn các mục.

l = [ 1, 4, 6, 30, 2, ... ]

Tôi muốn lấy số mục từ danh sách đó, trong đó một mục phải thỏa mãn điều kiện nhất định. Suy nghĩ đầu tiên của tôi là:

count = len([i for i in l if my_condition(l)])

Nhưng nếu danh sách được lọc my_condition () cũng có số lượng lớn các mục, tôi nghĩ rằng việc tạo danh sách mới cho kết quả được lọc chỉ lãng phí bộ nhớ. Đối với hiệu quả, IMHO, cuộc gọi trên không thể tốt hơn:

count = 0
for i in l:
    if my_condition(l):
        count += 1

Có cách nào theo kiểu chức năng để đạt được # mục thỏa mãn điều kiện nhất định mà không tạo danh sách tạm thời không?

Cảm ơn trước.

Bạn có thể sử dụng một biểu thức trình tạo :

>>> l = [1, 3, 7, 2, 6, 8, 10]
>>> sum(1 for i in l if i % 4 == 3)
2

hoặc thậm chí

>>> sum(i % 4 == 3 for i in l)
2

trong đó sử dụng thực tế rằng int(True) == 1.

Ngoài ra, bạn có thể sử dụng itertools.imap(python 2) hoặc đơn giản map(python 3):

>>> def my_condition(x):
...     return x % 4 == 3
... 
>>> sum(map(my_condition, l))
2

Bạn muốn hiểu rõ về trình tạo hơn là một danh sách ở đây.

Ví dụ,

l = [1, 4, 6, 7, 30, 2]

def my_condition(x):
    return x > 5 and x < 20

print sum(1 for x in l if my_condition(x))
# -> 2
print sum(1 for x in range(1000000) if my_condition(x))
# -> 14

Hoặc sử dụng itertools.imap(mặc dù tôi nghĩ rằng danh sách rõ ràng và các biểu thức trình tạo trông giống Pythonic hơn).

Lưu ý rằng, mặc dù nó không rõ ràng từ sumví dụ, nhưng bạn có thể soạn thảo phần hiểu của trình tạo một cách độc đáo. Ví dụ,

inputs = xrange(1000000)      # In Python 3 and above, use range instead of xrange
odds = (x for x in inputs if x % 2)  # Pick odd numbers
sq_inc = (x**2 + 1 for x in odds)    # Square and add one
print sum(x/2 for x in sq_inc)       # Actually evaluate each one
# -> 83333333333500000

Điều thú vị về kỹ thuật này là bạn có thể chỉ định các bước tách biệt về mặt khái niệm trong mã mà không buộc phải đánh giá và lưu trữ trong bộ nhớ cho đến khi kết quả cuối cùng được đánh giá.

Điều này cũng có thể được thực hiện bằng cách sử dụng reducenếu bạn thích lập trình chức năng

reduce(lambda count, i: count + my_condition(i), l, 0)

Bằng cách này, bạn chỉ thực hiện 1 lần và không có danh sách trung gian nào được tạo.

bạn có thể làm điều gì đó như:

l = [1,2,3,4,5,..]
count = sum(1 for i in l if my_condition(i))

mà chỉ thêm 1 cho mỗi phần tử thỏa mãn điều kiện.

from itertools import imap
sum(imap(my_condition, l))