从文本文件制作python字典

Question

man
  cat
  dog
apple
  sun
  friend

I have text file in above format I need to make python dictionary: {'man':'cat, dog', apple: 'sun', 'friend'} words man and apple have no spaces all the others have 2 spaces before so these two should be keys and words under them should be values.

input_file = "/home/yan/lem.txt"

class myDict(dict):

   def __init__(self):
       self = dict()

   def add(self, key, value):
       self[key] = value

lema = myDict()

with open(input_file, encoding='utf8', errors='ignore') as f:
    for line in f:
       if not line.startswith("  "):
           lema.add(line.split(), [i for i in f if i.startswith("  ")])
print(lema)

Was trying above code, but it gives not what I am expecting.

Answer 1

this will get you what you need:

d = {}
k = ''
with open('lem.txt') as f:
    for ln in f.readlines():
        if ln.startswith('  '):
            d.setdefault(k, []).append(ln.strip())
        else:
            k = ln.strip()

print d

Answer 2

1. you cannot use self as your dictionary name, in python it is used to refer your instance variable although it is not reserved keyword.

2. the class function add() , try use dict.setdefault(key, list()).append(value) it keeps appends new values to the existing key.

3. override __str__ method which display your expected string format when you invoke print(lema)

$ cat test.py

    input_file = "lem.txt"
    class myDict(dict):

       def __init__(self):
           self.mydict = dict()

       def add(self, key, value):
           self.mydict.setdefault(key, list()).append(value)

       def __str__(self):
           tmp = list()
           for key, value in self.mydict.items():
               tmp.append("{0}:{1}".format(key, value))
           return "\n".join(tmp)

    lema = myDict()

    with open(input_file) as f:
        for line in f:
           if not line.strip():
               '''skips blank line'''
               continue
           elif not line.startswith(" "):
               key = line.strip()
           elif line.startswith(" "):
               value = line.strip()
               lema.add(key,value)

    print(lema)

The output is:

    $python test.py
    apple:['sun', 'friend']
    man:['cat', 'dog']

Answer 3

You could slurp the file into one big string, s , and then do something like this:

s = """man
  cat
  dog
apple
  sun
  friend"""

d = {}

for line in s.replace("\n  ",',').split('\n'):
    words = line.split(',')
    d[words[0]] = words[1:]

>>> d
{'man': ['cat', 'dog'], 'apple': ['sun', 'friend']}

To get a feel for what is happening here:

>>> s.replace("\n  ",',')
'man,cat,dog\napple,sun,friend'