将此文件转换为字典

Question

我在 Stackoverflow 上环顾四周，但可以但找不到这里讨论的内容。我有一个要转换成字典的列表。

3
banana
apple
orange
5.00E-2
1
7.02E-4
1.00E-4
4
5.13E-4
-5.76E-2
8
7.23E-8
-6.12E-9
4
5.02E

数字 3 是键的数量。接下来的 3 个字符串是键名。接下来的 3 个值是列表的第一个元素....等等。例如我会得到：

  mydict['banana']=[ 5.00E-2, 1.00E-4, -5.76E-2, -6.12E-9]
  mydict['apple'] =[1,           4,       8,        4]
  mydict['orange']=[ 7.02E-4, 5.13E-4,  7.23E-8,   np.nan ]

请注意，最后一个元素并不完整，只是“5.02E”，必须用 numpy.nan 替换。此外，每个列表中的元素数量必须相同，缺失值应为 numpy.nan。这是一个 python 2.7.x 的问题。

所以我的问题是我正在寻找创建字典的最佳方式。怎么做？

Answer 1

num = mylist[0]
mydict = {}
for i in range(num):
    mydict[mylist[i + 1]] = mylist[i + 1 + num::num]

或者如果你更喜欢单行，

mydict = {mylist[i + 1]:mylist[num + i + 1::num] for i in range(mylist[0])}

请注意，这不会用numpy.nan 替换缺失的项目；就像

from itertools import izip_longest
import numpy as np

num      = mylist[0]
labels   = mylist[1:num + 1]
num_cols = (len(mylist) - 2) // num
cols     = (mylist[1 + c*num::num] for c in range(1, num_cols + 2))
rows     = izip_longest(*cols, fillvalue=np.nan)

mydict = dict(zip(labels, rows))

Answer 2

我认为，itertools.cycle + izip 可能在这里有用：

from collections import defaultdict
from itertools import cycle, izip, count, repeat

result = defaultdict(list)

with open(path) as fp:
    num = int(next(fp))
    idx = 0

    # create infinite loop of repeating keys to match them with the values:
    keys = cycle([next(fp).strip() for _ in range(num)])

    for k, v, c in izip(keys, fp, count()):
        result[k].append(np.float(v))

    if (c + 1) % num: # need to populate with nans
        result[k].extend(repeat(np.nan, num - (c + 1) % num))