如何拆分输入文件的第一行并将它们作为字典存储在 python 中？

Question

The first line of my input file looks like this:

<doc id="12" url="http://en.wikipedia.org/wiki?curid=12" title="Anarchism">

I want store them as key-value pair like this in python:

{doc_id: 12, url: http://en.wikipedia.org/wiki?curid=12, title: Anarchism} 

Here is my code:

infile=open('wiki_00').readline().rstrip()
infile.split()[1:]  

output looks like this:

['id="12"',
'url="http://en.wikipedia.org/wiki?curid=12"',
'title="Anarchism">']

But I would like the "", <> removed and id to be stored as type int

Answer 1

Don't do line[1:] to strip away the brackets. Use the strip method: line.strip(' <>') will remove all whitespace and <> characters from the ends of the line.

Something like this will do what I think you want. You may want to add error handling.

def turn_line_into_dict(line):
    # remove the brackets and tag name
    line = line.strip(' <>')
    first_space_idx = line.find(' ')
    line_without_tag = line[first_space_idx+1:]

    attr_list = line_without_tag.split(' ')

    d = {}
    for attr_str in attr_list :
       key,value = attr_str.split('=', 1) # only search for first occurrence, so an '=' in the url doesn't screw this up
       d[key] = value.strip('"\'') # remove quotes and let the dict figure out the type

    return d