我正在处理图像处理脚本。我需要让用户指定如何通过文本文件重新映射图像中的某些类。此文件中的语法应该简单明了。我想做的是让用户编写字典的字符串版本:
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
有没有一种简单的方法可以从“字符串版本”转换为真正的字典?您能想到用户可以使用的更简单/替代的单行语法吗?
答案 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'