itertools是Python中的一個模塊，具有用於處理迭代器的函數集合。它們使遍曆列表和字符串之類的可迭代對象變得非常容易。這樣的itertools函數之一是takewhile()。
注意：有關更多信息，請參閱Python Itertools。
takewhile()
這允許從迭代開始考慮一個項目，直到指定的謂詞首次變為假。在大多數情況下，迭代器是列表或字符串。顧名思義，“take”是序列“while”中的元素，謂詞是“true”。該函數屬於“terminating iterators”類別。輸出不能直接使用，而必須轉換為另一種可迭代的形式。通常，它們會轉換為列表。
用法:
takewhile(predicate, iterable)
的predicate是內置函數或用戶定義的函數。它甚至可以是lambda函數。
下麵給出了使用簡單if-else對該函數的一般實現。
def takewhile(predicate, iterable):
for i in iterable:
if predicate(i):
return(i)
else:
break
函數takewhile()以謂詞和可迭代作為參數。迭代iterable以檢查其每個元素。如果指定謂詞上的元素的值為true，則將其返回。否則，循環終止。
示例1：列表和takewhile()
考慮一個整數列表。我們隻需要輸出中的偶數。查看下麵的代碼，看看如果我們使用takewhile()。
from itertools import takewhile
# function to check whether
# number is even
def even_nos(x):
return(x % 2 == 0)
# iterable (list)
li =[0, 2, 4, 8, 22, 34, 6, 67]
# output list
new_li = list(takewhile(even_nos, li))
print(new_li)[0, 2, 4, 8, 22, 34, 6]
示例2：字符串和takewhile()
考慮一個alpha-numerical字符串。現在，我們需要將元素取為數字。
from itertools import takewhile
# function to test the elements
def test_func(x):
print("Testing:", x)
return(x.isdigit())
# using takewhile with for-loop
for i in takewhile(test_func, "11234erdg456"):
print("Output:", i)
print()輸出：
Testing:1 Output:1 Testing:1 Output:1 Testing:2 Output:2 Testing:3 Output:3 Testing:4 Output:4 Testing:e
可迭代對象也可以直接傳遞。將它們傳遞給某些變量不是強製性的 takewhile()函數。
示例3：takewhile()中的lambda函數
考慮輸入字符串的元素，直到遇到“ s”為止。
from itertools import takewhile
# input string
st ="GeeksforGeeks"
# consider elements until
# 's' is encountered
li = list(takewhile(lambda x:x !='s', st))
print(li)輸出：
['G', 'e', 'e', 'k']
範例4：
按隨機順序列出字母，並考慮元素，直到遇到“ e”，“ i”或“ u”。
import random
from itertools import takewhile
# generating alphabets in random order
li = random.sample(range(97, 123), 26)
li = list(map(chr, li))
print("The original list list is:")
print(li)
# consider the element until
# 'e' or 'i' or 'o' is encountered
new_li = list(takewhile(lambda x:x not in ['e', 'i', 'o'],
li))
print("\nThe new list is:")
print(new_li)The original list list is:
[‘v’, ‘u’, ‘k’, ‘j’, ‘r’, ‘q’, ‘n’, ‘y’, ‘a’, ‘x’, ‘i’, ‘p’, ‘e’, ‘w’, ‘b’, ‘t’, ‘s’, ‘l’, ‘z’, ‘m’, ‘f’, ‘c’, ‘g’, ‘d’, ‘o’, ‘h’]The new list is:
[‘v’, ‘u’, ‘k’, ‘j’, ‘r’, ‘q’, ‘n’, ‘y’, ‘a’, ‘x’]
相關用法
注：本文由純淨天空篩選整理自erakshaya485大神的英文原創作品 Python – Itertools.takewhile()。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。