2009-10-10 44 views
3

在Python中,我需要生成一个dict,它将一个字母映射到该字母的预定义“one-hot”表示形式。通过图示的方式,dict应该是这样的:生成单热编码的字符串表示形式

{ 'A': '1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0', 
    'B': '0 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0', # ... 
} 

有每字母表的字母一个位(表示为一个字符)。因此,每个字符串将包含25个零和一个1. 1的位置由字母表中相应字母的位置决定。

我想出了这个生成一些代码:

# Character set is explicitly specified for fine grained control 
_letters = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" 
n = len(_letters) 
one_hot = [' '.join(['0']*a + ['1'] + ['0']*b) 
      for a, b in zip(range(n), range(n-1, -1, -1))] 
outputs = dict(zip(_letters, one_hot)) 

是否有更有效的/清洁/更Python的方式做同样的事情?

回答

7

我觉得这是更具可读性:

from string import ascii_uppercase 

one_hot = {} 
for i, l in enumerate(ascii_uppercase): 
    bits = ['0']*26; bits[i] = '1' 
    one_hot[l] = ' '.join(bits) 

如果你需要一个更一般的字母,只是列举过的字符串,并与取代['0']*26

+0

可读性很强。 +1 – 2009-10-10 20:43:21

+0

不错。我想我在函数式编程中尝试了一个丑陋的尝试。尽管你的版本更易读。 – 2009-10-10 20:51:54

+0

'ascii_uppercase'的使用是一个很好的接触(我甚至不知道它存在),但我不确定这比原始代码更具可读性。列表理解简洁,但也非常有表现力。使用明确的循环使得代码的目的在我眼中不太清晰。 – bcat 2009-10-10 20:57:58

0

这似乎很清楚,简洁,Pythonic对我来说。

1
one_hot = [' '.join(['0']*a + ['1'] + ['0']*b) 
      for a, b in zip(range(n), range(n-1, -1, -1))] 
outputs = dict(zip(_letters, one_hot)) 

特别是,有一个代码很多挤进这两条线。您可以尝试Introduce Explaining Variable重构。或者可能是extract method

这里有一个例子:

def single_onehot(a, b): 
    return ' '.join(['0']*a + ['1'] + ['0']*b) 

range_zip = zip(range(n), range(n-1, -1, -1)) 
one_hot = [ single_onehot(a, b) for a, b in range_zip] 
outputs = dict(zip(_letters, one_hot)) 

虽然你可能与我的不同意命名。

2

在Python 2.5和最多可以使用条件运算符:

from string import ascii_uppercase 

one_hot = {} 
for i, c in enumerate(ascii_uppercase): 
    one_hot[c] = ' '.join('1' if j == i else '0' for j in range(26))