Python-从numpy表创建字典以显示列表

时间:2019-01-21 21:18:39

标签: python numpy dictionary

给出以下numpy表

GR = [
    ['Student', 'Exam 1', 'Exam 2', 'Exam 3'],
    ['PersonA', '100', '90', '80'],
    ['PersonB', '88', '99', '111'],
    ['PersonC', '45', '56', '67'],
    ['PersonD', '59', '61', '67'],
    ['PersonE', '73', '79', '83'],
    ['PersonF', '89', '97', '101']
     ]

我需要创建一个名为GL的字典,该字典将学生的姓名映射到他们的考试成绩列表。成绩应从str转换为int。

Desired Output: {'PersonA':[100, 90, 80], 'PersonB':[88, 99, 111], ect....}

5 个答案:

答案 0 :(得分:1)

我在理解上有些麻烦,但是我认为这就是您想要的:

GL = [
    {x[0]: [int(j) for j in x[1:]]} for x in GR[1:]
]

输出:

[{'PersonA': [100, 90, 80]},
 {'PersonB': [88, 99, 111]},
 {'PersonC': [45, 56, 67]},
 {'PersonD': [59, 61, 67]},
 {'PersonE': [73, 79, 83]},
 {'PersonF': [89, 97, 101]}]

答案 1 :(得分:1)

Python 3的另一种解决方案,使用map将等级从str转换为int,并进行部分拆包以将每个列表分成person和它们的{ {1}}

grades

等效于Python 2

In [111]: {person: list(map(int, grades)) for person, *grades in GR[1:]}
Out[111]:
{'PersonA': [100, 90, 80],
 'PersonB': [88, 99, 111],
 'PersonC': [45, 56, 67],
 'PersonD': [59, 61, 67],
 'PersonE': [73, 79, 83],
 'PersonF': [89, 97, 101]}

答案 2 :(得分:0)

选项1:

GR_ = {i[0]: list(map(int, i[1:])) for i in GR[1:]}

选项2:

GR_ = dict(map(lambda x: (x[0], list(map(int, x[1:]))), GR[1:]))

输出:

{'PersonA': [100, 90, 80],
 'PersonB': [88, 99, 111],
 'PersonC': [45, 56, 67],
 'PersonD': [59, 61, 67],
 'PersonE': [73, 79, 83],
 'PersonF': [89, 97, 101]}

答案 3 :(得分:0)

您的GR是一个numpy数组(使用a[2,0]索引可确认这一点)

In [179]: GR
Out[179]: 
array([['Student', 'Exam 1', 'Exam 2', 'Exam 3'],
       ['PersonA', '100', '90', '80'],
       ['PersonB', '88', '99', '111'],
       ['PersonC', '45', '56', '67'],
       ['PersonD', '59', '61', '67'],
       ['PersonE', '73', '79', '83'],
       ['PersonF', '89', '97', '101']], dtype='<U7')

所以您正在做的是:

In [185]: {row[0]: row[1:] for row in GR[1:]}
Out[185]: 
{'PersonA': array(['100', '90', '80'], dtype='<U7'),
 'PersonB': array(['88', '99', '111'], dtype='<U7'),
 'PersonC': array(['45', '56', '67'], dtype='<U7'),
 'PersonD': array(['59', '61', '67'], dtype='<U7'),
 'PersonE': array(['73', '79', '83'], dtype='<U7'),
 'PersonF': array(['89', '97', '101'], dtype='<U7')}

使用以下命令可以轻松地将字符串数组转换为整数:

In [186]: {row[0]: row[1:].astype(int) for row in GR[1:]}
Out[186]: 
{'PersonA': array([100,  90,  80]),
 'PersonB': array([ 88,  99, 111]),
 'PersonC': array([45, 56, 67]),
 'PersonD': array([59, 61, 67]),
 'PersonE': array([73, 79, 83]),
 'PersonF': array([ 89,  97, 101])}

以及列表:

In [187]: {row[0]: row[1:].astype(int).tolist() for row in GR[1:]}
Out[187]: 
{'PersonA': [100, 90, 80],
 'PersonB': [88, 99, 111],
 'PersonC': [45, 56, 67],
 'PersonD': [59, 61, 67],
 'PersonE': [73, 79, 83],
 'PersonF': [89, 97, 101]}

我对列表转换的提名是:

In [188]: {name: [int(i) for i in grades] for name, *grades in GR[1:]}
Out[188]: 
{'PersonA': [100, 90, 80],
 'PersonB': [88, 99, 111],
 'PersonC': [45, 56, 67],
 'PersonD': [59, 61, 67],
 'PersonE': [73, 79, 83],
 'PersonF': [89, 97, 101]}

答案 4 :(得分:0)

我很抱歉@Charles Landau,自从提出这个问题以来,我的python技能得到了极大的提高。

GL = dict() 
     for i in GR[1:]:
          GL[i[0]] = [int(i[1]),int(i[2]),int(i[3])]
     print(GL)