protein-retrieval-multimodal / utils /convert_str_to_list.py
lindsay-qu's picture
Create convert_str_to_list.py
29d514e verified
raw
history blame
676 Bytes
import ast
def convert_str_to_list(str_input):
try:
# 使用ast.literal_eval安全地将字符串转换为列表
output = ast.literal_eval(str_input)
return ast.literal_eval(str_input)
except (ValueError, SyntaxError):
if str_input[0] == '[' and str_input[1] == '[':
output = str_input.lstrip('[').rstrip(']').replace('], [', '],[').split('],[')
output = [o.replace('"', "'").replace("\n", "").replace("', '", "','").split("','") for o in output]
else:
output = str_input.replace('"', "'").replace("\n", "").replace("', '", "','").lstrip("['").rstrip("']").split("','")
return output