在python中从文件创建字典

Question

例如，我有像这样的多行文件

<<something>>   1, 5, 8
<<somethingelse>> hello
<<somethingelseelse>> 1,5,6

我需要用键创建字典

dict = { "something":[1,5,8], "somethingelse": "hello" ...}

我需要以某种方式读取 > 里面的内容并将其作为键，并且我还需要检查是否有很多元素或只有 1 个。如果只有一个，那么我将它作为字符串。如果不止一个，那么我需要将其作为元素列表。 任何想法如何帮助我？ 也许是正则表达式，但我不太喜欢它们。

我很容易创建 def 正在读取文件行，但不知道如何分隔这些值：

f = open('something.txt', 'r')
lines = f.readlines()
f.close()

def finding_path():
    for line in lines:
        print line

finding_path()
f.close()

有什么想法吗？谢谢:)

Answer 1

假设您的键始终是单个单词，您可以使用split(char, maxSplits)。类似下面的东西

import sys

def finding_path(file_name):
    f = open(file_name, 'r')
    my_dict = {}
    for line in f:
        # split on first occurance of space
        key_val_pair = line.split(' ', 1)
        # if we do have a key seprated by a space
        if len(key_val_pair) > 1:
            key = key_val_pair[0]
            # proceed only if the key is enclosed within '<<' and '>>'
            if key.startswith('<<') and key.endswith('>>'):
                key = key[2:-2]
                # put more than one value in list, otherwise directly a string literal
                val = key_val_pair[1].split(',') if ',' in key_val_pair[1] else key_val_pair[1]

                my_dict[key] = val
    print my_dict
    f.close()

if __name__ == '__main__':
    finding_path(sys.argv[1])

使用如下文件

<<one>> 1, 5, 8
<<two>> hello
// this is a comment, it will be skipped
<<three>> 1,5,6

我得到输出

{'three': ['1', '5', '6\n'], 'two': 'hello\n', 'one': ['1', ' 5', ' 8\n']}

Answer 2

请检查以下代码：

• 使用正则表达式获取键和值

• 如果值列表的长度为1，则将其转换为字符串。

import re
demo_dict = {}

with open("val.txt",'r') as f:
    for line in f:
          m= re.search(r"<<(.*?)>>(.*)",line)
          if m is not None:
               k = m.group(1)
               v = m.group(2).strip().split(',')
               if len(v) == 1:
                    v = v[0]
               demo_dict[k]=v
print demo_dict

输出：

C:\Users\dinesh_pundkar\Desktop>python demo.Py
{'somethingelseelse': [' 1', '5', '6'], 'somethingelse': 'hello', 'something': [
'   1', ' 5', ' 8']}

Answer 3

我的回答与 Dinesh 的类似。如果可能的话，我添加了一个函数来将列表中的值转换为数字，并添加了一些错误处理，以便如果行不匹配，则会给出有用的警告。

import re
import warnings

regexp =re.compile(r'<<(\w+)>>\s+(.*)')

lines = ["<<something>>   1, 5, 8\n",
         "<<somethingelse>> hello\n",
         "<<somethingelseelse>> 1,5,6\n"]

#In real use use a file descriptor instead of the list
#lines = open('something.txt','r')

def get_value(obj):
    """Converts an object to a number if possible, 
    or a string if not possible"""
    try:
        return int(obj)
    except ValueError:
        pass
    try:
        return float(obj)
    except ValueError:
        return str(obj)

dictionary = {}

for line in lines:    
    line = line.strip()
    m = re.search(regexp, line)
    if m is None:
        warnings.warn("Match failed on \n   {}".format(line))
        continue
    key = m.group(1)
    value = [get_value(x) for x in m.group(2).split(',')]
    if len(value) == 1: 
        value = value[0]
    dictionary[key] = value

print(dictionary)

输出

{'something': [1, 5, 8], 'somethingelse': 'hello', 'somethingelseelse': [1, 5, 6]}