将字典的字符串表示转换为真实字典

时间:2010-11-08 19:37:42

标签: python string dictionary numpy

我正在处理图像处理脚本。我需要让用户指定如何通过文本文件重新映射图像中的某些类。此文件中的语法应该简单明了。我想做的是让用户编写字典的字符串版本:

125:126, 126:126, 127:128, 128:128

然后将其转换为真正的字典(这是缺少的链接):

a = {125:126, 126:126, 127:128, 128:128}

然后将重新映射图像类,如下所示:

u, indices = numpy.unique(image, return_inverse=True)
for i in range(0, len(u)):
    u[i] = a[u[i]]
updatedimage = u[indices]
updatedimage = numpy.resize(updatedimage, (height, width)) #Resize to original dims

有没有一种简单的方法可以从“字符串版本”转换为真正的字典?您能想到用户可以使用的更简单/替代的单行语法吗?

1 个答案:

答案 0 :(得分:11)

您可以使用ast.literal_eval

>>> import ast
>>> ast.literal_eval('{' + s + '}')
{128: 128, 125: 126, 126: 126, 127: 128}

请注意,这需要Python 2.6或更高版本。

另一种方法是将字符串拆分为',',然后将':'上的每个字符拆分并从中构建dict

>>> dict(map(int, x.split(':')) for x in s.split(','))
{128: 128, 125: 126, 126: 126, 127: 128}

两个示例都假设您的初始字符串位于名为s的变量中:

>>> s = '125:126, 126:126, 127:128, 128:128'