преобразовать строку в dict, используя понимание списка - PullRequest
15 голосов
/ 07 августа 2009

Я сталкивался с этой проблемой несколько раз и, похоже, не могу найти простое решение. Скажи, у меня есть строка

string = "a=0 b=1 c=3"

Я хочу преобразовать это в словарь с ключами a, b и c, а 0, 1 и 3 - их соответствующими значениями (преобразованными в int) Очевидно, я могу сделать это:

list = string.split()
dic = {}
for entry in list:
    key, val = entry.split('=')
    dic[key] = int(val)

Но мне не очень нравится этот цикл, он кажется настолько простым, что вы сможете преобразовать его в какое-то выражение для понимания списка. И это работает для немного более простых случаев, когда val может быть строкой.

dic = dict([entry.split('=') for entry in list])

Однако мне нужно конвертировать val в int на лету, и делать что-то подобное синтаксически некорректно.

dic = dict([[entry[0], int(entry[1])] for entry.split('=') in list])

Итак, мой вопрос: есть ли способ устранить цикл for, используя понимание списка? Если нет, то есть ли какой-нибудь встроенный метод python, который сделает это для меня?

Ответы [ 8 ]

29 голосов
/ 07 августа 2009

Ты имеешь в виду это?

>>> dict( (n,int(v)) for n,v in (a.split('=') for a in string.split() ) )
{'a': 0, 'c': 3, 'b': 1}
3 голосов
/ 08 августа 2009

Попробуйте следующее:

dict([x.split('=') for x in s.split()])
3 голосов
/ 08 августа 2009

Как насчет однострочника без понимания списка?

 foo="a=0 b=1 c=3"
 ans=eval( 'dict(%s)'%foo.replace(' ',',')) )
 print ans
{'a': 0, 'c': 3, 'b': 1}
2 голосов
/ 20 ноября 2014

В настоящее время вам, вероятно, следует использовать словарное понимание, представленное в 2.7:

mydict = {key: int(value) for key, value in (a.split('=') for a in mystring.split())}

Понимание словаря быстрее и сияет (и, на мой взгляд, более читабельно).

from timeit import timeit

setup = """mystring = "a=0 b=1 c=3\""""
code1 = """mydict = dict((n,int(v)) for n,v in (a.split('=') for a in mystring.split()))""" # S.Lott's code
code2 = """mydict = {key: int(value) for key, value in (a.split('=') for a in mystring.split())}"""

print timeit(code1, setup=setup, number=10000) # 0.115524053574
print timeit(code2, setup=setup, number=10000) # 0.105328798294
2 голосов
/ 08 августа 2009

Мне иногда нравится такой подход, особенно когда логика создания ключей и значений более сложна:

s = "a=0 b=1 c=3"

def get_key_val(x):
    a,b = x.split('=')
    return a,int(b)

ans = dict(map(get_key_val,s.split()))
1 голос
/ 07 августа 2009
from cgi import parse_qsl
text = "a=0 b=1 c=3"
dic = dict((k, int(v)) for k, v in parse_qsl(text.replace(' ', '&')))
print dic

печать

{'a': 0, 'c': 3, 'b': 1}
0 голосов
/ 13 июня 2010

Я бы сделал это:

def kv(e): return (e[0], int(e[1]))
d = dict([kv(e.split("=")) for e in string.split(" ")])
0 голосов
/ 08 августа 2009

Мне нравится решение С. Лотта, но я предложил другую возможность.
Поскольку у вас уже есть строка, похожая на то, как вы ее пишете, вы можете просто адаптировать ее к синтаксису python и затем выполнить eval () it:)

import re
string = "a=0 b=1 c=3"
string2 = "{"+ re.sub('( |^)(?P<id>\w+)=(?P<val>\d+)', ' "\g<id>":\g<val>,', string) +"}"
dict = eval(string2)
print type(string), type(string2), type(dict)
print string, string2, dict

Регулярное выражение здесь довольно сырое и не будет перехватывать все возможные идентификаторы python, но я хотел сделать его простым для простоты. Конечно, если у вас есть контроль над тем, как генерируется входная строка, просто сгенерируйте ее в соответствии с синтаксисом Python и оцените ее. НО, конечно, вы должны выполнить дополнительные проверки работоспособности, чтобы убедиться, что код там не введен!

...