2015-06-03 78 views
2

我有一个字典与字符串和2元组键。我想将(x,y)中的所有2元组键转换为x:y的字符串。这里是我的数据:只选择元组的字典键?

In [4]: 

data = {('category1', 'category2'): {'numeric_float1': {('Green', 'Car'): 0.51376354561039017,('Red', 'Plane'): 0.42304110216698415,('Yellow', 'Boat'): 0.56792298947973241}}} 
data 
Out[4]: 
{('category1', 
    'category2'): {'numeric_float1': {('Green', 'Car'): 0.5137635456103902, 
    ('Red', 'Plane'): 0.42304110216698415, 
    ('Yellow', 'Boat'): 0.5679229894797324}}} 

然而,这是字典输出我想:创建一个递归函数,改变所有的按键

{'category1:category2': 
    {'numeric_float1': 
     {'Green:Car': 0.5137635456103902, 
     'Red:Plane': 0.42304110216698415, 
     'Yellow:Boat': 0.5679229894797324}}} 

我修改代码a previous SO answer

In [5]: 

def convert_keys_to_string(dictionary): 
    if not isinstance(dictionary, dict): 
     return dictionary 
    return dict((':'.join(k), convert_keys_to_string(v)) for k, v in dictionary.items()) 

convert_keys_to_string(data) 

但是我无法获得避免非元组键的函数。因为它没有避免非元组键,功能修复的2元组密钥,但打乱了非元组键:

Out[5]: 
{'category1:category2': {'n:u:m:e:r:i:c:_:f:l:o:a:t:1': {'Green:Car': 0.5137635456103902, 
    'Red:Plane': 0.42304110216698415, 
    'Yellow:Boat': 0.5679229894797324}}} 

回答

3

变化':'.join(k)k if hasattr(k, 'isalpha') else ':'.join(k)。如果它具有属性isalpha,这意味着它可能是一个字符串,或者用冒号连接对象,否则这将使用未改变的对象。或者(谢谢,@Padraic),您可以使用':'.join(k) if isinstance(k, tuple) else k

+1

我想喂鸭类型,但。 – TigerhawkT3

+0

我不遵循,如果hasattr(k,'isalpha')'做了issinstance(k,str)'不会呢? –

+0

我不是在问是否是鸭子,而是在聆听庸医。 :) – TigerhawkT3

1

你只关心类型的字典和元组所以只检查这两个递归的值:

def rec(d): 
    for k,v in d.items(): 
     if isinstance(v, dict): 
      rec(v) 
     if isinstance(k, tuple): 
      del d[k] 
      d[":".join(k)] = v 

rec(data) 

from pprint import pprint as pp 
pp(data) 

输出:

{'category1:category2': {'numeric_float1': {'Green:Car': 0.5137635456103902, 
              'Red:Plane': 0.42304110216698415, 
              'Yellow:Boat': 0.5679229894797324}}} 

此修改,我认为是实际的目标原来字典。

如果你希望它为所有iterables工作,除了一个STR:

from collections import Iterable 
def rec(d): 
    for k, v in d.items(): 
     if isinstance(v, dict): 
      rec(v) 
     if isinstance(k, Iterable) and not isinstance(k, str): 
      del d[k] 
      d[":".join(k)] = v 
1

通过@TigerhawkT3's answer的启发,这里是有点的“嘎嘎监听器”:

[':'.join(k), k][k in k] 

您可以使用替代您的无条件':'.join(k)。其他的想法:

[':'.join(k), k][''.join(k) == k] 
[':'.join(k), k][str(k) == k] 

我应该说,这些都是令人困惑,做不必要的工作,虽然。这只是为了好玩/打高尔夫球。 ... if isinstance(...) else ...是正确的方法。虽然,k in k实际上可能比isinstance(k, str)快:

>>> timeit('k in k',    "k = 'numeric_float1'") 
0.222242249806186 
>>> timeit('isinstance(k, str)', "k = 'numeric_float1'") 
0.3160444680784167 

>>> timeit('k in k',    "k = ('Yellow', 'Boat')") 
0.21133306092963267 
>>> timeit('isinstance(k, str)', "k = ('Yellow', 'Boat')") 
0.5903861610393051 
+0

为什么downvote?它有效,我警告它的坏处,我认为这是教育/有趣的。 –