按每個元素的類型過濾列表的元素

Question

我有不同類型的數據（字符串，int等）的列表。我需要創建一個新的列表，例如只有int元素，另一個列表只包含字符串元素。怎麼做？

Answer 1

你可以用list comprehension做到這一點：

integers = [elm for elm in data if isinstance(elm, int)] 

哪裏data是數據。以上所做的是創建一個新列表，使用滿足if之後的條件的元素data（elm）填充它，該元素將檢查元素是否爲int的實例。您還可以使用filter：

integers = list(filter(lambda elm: isinstance(elm, int), data)) 

以上會過濾掉根據傳遞的拉姆達，過濾掉所有非整數元素。然後您可以將其應用於字符串，使用isinstance(elm, str)來檢查字符串的實例。

Answer 2

排序列表按類型，然後使用groupby到組它：

>>> import itertools 
>>> l = ['a', 1, 2, 'b', 'e', 9.2, 'l'] 
>>> l.sort(key=lambda x: str(type(x))) 
>>> lists = [list(v) for k,v in itertools.groupby(l, lambda x: str(type(x)))] 
>>> lists 
[[9.2], [1, 2], ['a', 'b', 'e', 'l']]