2012-11-17 83 views
1

我有以下两个列表:的Python:比较列表和在共同领域的结合

ISO3166_CountryCodes_NO = [["NO","Norge"],["SE","Sverige"],["GR","Hellas"]] 
ISO3166_CountryCodes_EN = [["NO","Norway"],["SE","Sweden"],["GR","Greece"]] 

正如你看到的,国家代码始终是相同的,但该国名称不同(不同的译本) 如何能我创建一个列表如下:

ISO3166_CountryCodes = [["NO","Norge","Norway"],["SE","Sverige","Sweden"],["GR","Hellas","Greece"]] 

我可以用它做在第一列表循环,对于每个元素我可以搜索到第二个,找到国家代码是常见的。然后将译文附加到新列表中,但我觉得这种方式有点笨拙。

有没有更好的方法在Python中实现这一点?例如在我更熟悉的Perl中,我会使用一个哈希表。

回答

4

在Python中,词典是一个哈希表。首先,创建两个字典:

NO_dict = {x[0]: x[1] for x in ISO3166_CountryCodes_NO} 
EN_dict = {x[0]: x[1] for x in ISO3166_CountryCodes_EN} 

它给你:

{'GR': 'Hellas', 'NO': 'Norge', 'SE': 'Sverige'} 
{'GR': 'Greece', 'NO': 'Norway', 'SE': 'Sweden'} 

然后,您可以创建一个列表,像这样:

final_list = [[k, NO_dict[k], EN_dict[k]] for k in NO_dict] 

给你:

[['GR', 'Hellas', 'Greece'], 
['SE', 'Sverige', 'Sweden'], 
['NO', 'Norge', 'Norway']] 

你可能会稍后发现保持这种状态与

final_dict = {k:(NO_dict[k], EN_dict[k]) for k in NO_dict} 

所以,你可以使用缩写为关键例如抓取物品存放在元组的名称,例如:a词典ê数据final_dict['NO']会产生('Norge', 'Norway')

编辑:OrderedDict

如果你有蟒蛇> = 2.7,而你所关心的顺序,你仍然可以通过OrderedDict,例如使用字典:

from collections import OrderedDict 

# A list of lists can be used as input for an OrderedDict, so don't need to loop 
NO_dict = OrderedDict(ISO3166_CountryCodes_NO) 
EN_dict = OrderedDict(ISO3166_CountryCodes_EN) 

# Assumes you want the result in the same order as the Norwegian list 
# Iterate over the English list if it has a preferred order 

final_dict = OrderedDict([(k, (NO_dict[k], EN_dict[k])) for k in NO_dict]) 

(关于另一个实现,请参阅AshwiniChaudhary的答案)

+2

请注意,这个答案和Ashwini的答案在涉及允许的输入和结果输出时有两个区别:即使国家/地区的顺序在列表之间不同,但此方案(自然地)也不保留此顺序解决方案。 –

+0

谢谢!非常干净,简短的代码和易于理解:) –

+1

@lazyr增加了一个解决方案,即使列表是随机排列的,也保留了顺序。 –

2

这样的事情,使用unique_everseenitertools recipeschain()

In [26]: from itertools import * 

In [27]: lis1=[["NO","Norge"],["SE","Sverige"],["GR","Hellas"]] 

In [28]: lis2=[["NO","Norway"],["SE","Sweden"],["GR","Greece"]] 

In [29]: from itertools import * 

In [30]: def unique_everseen(iterable, key=None): 
     seen = set() 
     seen_add = seen.add 
     if key is None: 
       for element in ifilterfalse(seen.__contains__, iterable): 
         seen_add(element) 
         yield element 
       else: 
         for element in iterable: 
           k = key(element) 
           if k not in seen: 
             seen_add(k) 
             yield element 
    ....:       

In [31]: [list(unique_everseen(chain(*x))) for x in izip(lis1,lis2)] 
Out[31]: 
[['NO', 'Norge', 'Norway'], 
['SE', 'Sverige', 'Sweden'], 
['GR', 'Hellas', 'Greece']] 

或:可以使用groupby从itertools,与operator.itemgetter()结合:

In [42]: from operator import * 

In [43]: [[k]+list(map(itemgetter(1),g)) for x in zip(lis1,lis2) for k,g in groupby(x,itemgetter(0))] 
Out[43]: 
[['NO', 'Norge', 'Norway'], 
['SE', 'Sverige', 'Sweden'], 
['GR', 'Hellas', 'Greece']] 

或使用collections.OrderedDict,这是dict的子类别并维护顺序:

In [47]: from collections import OrderedDict 

In [48]: dic=OrderedDict() 

In [49]: for x in lis1: 
    ....:  dic.setdefault(x[0],[]).append(x[1]) 
    ....:  

In [50]: for x in lis2: 
    dic.setdefault(x[0],[]).append(x[1]) 
    ....:  

In [51]: dic 
Out[51]: OrderedDict([('NO', ['Norge', 'Norway']), ('SE', ['Sverige', 'Sweden']), ('GR', ['Hellas', 'Greece'])]) 

In [52]: [[x]+y for x,y in dic.items()] 
Out[52]: 
[['NO', 'Norge', 'Norway'], 
['SE', 'Sverige', 'Sweden'], 
['GR', 'Hellas', 'Greece']] 

#or directly access the names using the short-name 
In [53]: dic['NO'] 
Out[53]: ['Norge', 'Norway'] 

In [54]: dic['GR'] 
Out[54]: ['Hellas', 'Greece'] 
+0

谢谢你的许多选择! –

+0

@VangelisTasoulas不客气,你可以upvote或标记答案为接受,如果你觉得有帮助,我认为这是正确的方式说SO感谢。 –

1

你可以使用列表理解:

>>> [[s]+ 
    [n for (c,n) in ISO3166_CountryCodes_NO if c==s]+ 
    [n for (c,n) in ISO3166_CountryCodes_EN if c==s] 
    for s in set([c for (c,n) in ISO3166_CountryCodes_NO] + 
        [c for (c,n) in ISO3166_CountryCodes_EN])] 

[['GR', 'Hellas', 'Greece'], ['SE', 'Sverige', 'Sweden'], ['NO', 'Norge', 'Norway']] 
1

使用Python 3.2。

1路:

[[i[0],i[1],v[1]] for i in list1 for v in list2 if i[0]==v[0]] 

第二个办法:

res=[] 
for i,v in list(zip(list1,list2): 
    tem=[i[0]] 
    if i[0]==v[0]: tem.extend([i[1],v[1]]) 
res.append(tem) 
+0

请仅使用代码段的代码格式,而不是您的整个答案。有关更多格式选项,请参阅http://stackoverflow.com/editing-help –