檢查，如果項目是在列表中，然後返回該項目

Question

我有這樣的結構：

['a;1,2,3\n', 'b;abc\n', ...] 

換句話說

：這是一個List像這樣的項目：'id;element1,element2,...\n'

現在我想檢查列表包含了ID = "b"一個元素，如果它cotains項目"b"我想回到整個元素：

'b;abc\n' 

如何用python做到這一點？是否有可能通過in聲明來做到這一點？

Answer 1

這將返回符合條件的所有物品的清單，我認爲它可能有不止一個相匹配的結果，如果只有1條結果，結果列表將有1項。

>>> input = ['a;1,2,3\n', 'b;abc\n'] 
>>> filter(lambda item:item.find('b;') == 0 ,input) 
['b;abc\n'] 

Answer 2

>>> L = ['a;1,2,3\n', 'b;abc\n'] 
>>> next((x for x in L if x.partition(';')[0] == 'b'), 'No match') 
'b;abc\n' 

Answer 3

您可以使用下面的函數來提取您的結構匹配的所有元素的列表：

def query(data, key): 
    return [ x for x in data if x.startswith(key + ';') ] 

使用例：

data = [ 
    'a;1,2,3\n', 
    'b;abc\n', 
    'c;4,5,6\n' 
] 
print query(data, 'b') 

這給列表。如果你要求一個不存在的密鑰，結果是空的列表。

Answer 4

其他方法最有可能更快。

>>> L = ['a;1,2,3\n', 'b;abc\n'] 
>>> temp = [i.split(';') for i in L] 
>>> ';'.join([x for x in temp if x[0] == 'b'][0]) 
'b;abc\n'