集合不可散列,因为它们是可变的。但是有没有办法仍然在 {1, 2, {3, 4}} 上使用literal_eval?我只想知道外部结构是一个集合,我不关心内部类型,但集合内的集合是可能的输入。
更新:
输入是从文件中以字符串形式读取的。
最佳答案
您可以破解ast.literal_eval
,使其在看到集合时返回卡住集。具体做法如下:
- 搜索您的 Python 安装库所在的位置
- 它包含文件
ast.py
,该文件包含函数literal_eval
- 在您自己的模块中复制该函数(使用不同的名称)并将其更改为从
ast
模块导入所有相关名称 - 在处理
Set
的行中,将set
的生成替换为frozenset
然后您可以使用它来安全地解析包含集合的文字集。对于我的 Python 3.5,我使用:
def frozen_literal_eval(node_or_string):
"""
Safely evaluate an expression node or a string containing a Python
expression. The string or node provided may only consist of the following
Python literal structures: strings, bytes, numbers, tuples, lists, dicts,
sets, booleans, and None.
SPECIAL: This version uses frozensets instead of sets
"""
# SPECIAL: import names from ast module
from ast import parse, Expression, Str, Bytes, Num, Tuple, List, Set, Dict
from ast import NameConstant, UnaryOp, UAdd, USub, BinOp, Add, Sub
# END SPECIAL
if isinstance(node_or_string, str):
node_or_string = parse(node_or_string, mode='eval')
if isinstance(node_or_string, Expression):
node_or_string = node_or_string.body
def _convert(node):
if isinstance(node, (Str, Bytes)):
return node.s
elif isinstance(node, Num):
return node.n
elif isinstance(node, Tuple):
return tuple(map(_convert, node.elts))
elif isinstance(node, List):
return list(map(_convert, node.elts))
elif isinstance(node, Set):
#SPECIAL: returns a frozenset
return frozenset(map(_convert, node.elts))
# END SPECIAL
elif isinstance(node, Dict):
return dict((_convert(k), _convert(v)) for k, v
in zip(node.keys, node.values))
elif isinstance(node, NameConstant):
return node.value
elif isinstance(node, UnaryOp) and \
isinstance(node.op, (UAdd, USub)) and \
isinstance(node.operand, (Num, UnaryOp, BinOp)):
operand = _convert(node.operand)
if isinstance(node.op, UAdd):
return + operand
else:
return - operand
elif isinstance(node, BinOp) and \
isinstance(node.op, (Add, Sub)) and \
isinstance(node.right, (Num, UnaryOp, BinOp)) and \
isinstance(node.left, (Num, UnaryOp, BinOp)):
left = _convert(node.left)
right = _convert(node.right)
if isinstance(node.op, Add):
return left + right
else:
return left - right
raise ValueError('malformed node or string: ' + repr(node))
return _convert(node_or_string)
我可以使用:
>>> s = '{ 1, 2, {3, 4}}'
>>> frozen_literal_eval(s)
frozenset({1, 2, frozenset({3, 4})})
关于Python:如何对集合中的集合使用ast.literal_eval()?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41348456/