我需要将值的字典中的前导字符切掉 - 但前提是值的长度大于 1。目前我正在使用字典理解来执行此操作:
new_dict = {item[0]:item[1][1:] for item in old_dict if item.startswith('1')}
但我不知道如何修改它,以便单独保留长度为 1 的键。
密钥是霍夫曼代码的代码字,因此以“0”或“1”开头。
一个示例代码是:
code = {'a':'0', 'b':'10', 'c':'110', 'd':'111'}
上面的代码对 'b','c','d' 工作正常,但对 'a' 失败(这是内涵 - 这是一个单元测试)。
如何正确修改上述示例以通过测试?