python3字典,值作为列表

时间:2015-07-12 05:41:37

标签: python list python-2.7 dictionary python-3.4

以下代码在Python 2.7中没有任何问题,但在Python 3.4.3中,抛出TypeError: iteration over a 0-d array

import numpy as np
def c1():
        mat=[]
        mat.append(np.array(map(float,np.random.rand(7))))
        return mat
mylist=[]
mydict_={}
mydict_['dog']=c1()
mydict_['big']=c1()

justKeys={label:1 for label in list(mydict_.keys())}

for idx, word in enumerate(justKeys.keys()):
        mylist.append(np.array(map(float, mydict_[word][0])))

对于python2.7和3.4中的列表mylist,我应该怎么做才能获得相同的结果?

1 个答案:

答案 0 :(得分:2)

问题不在于字典方法,而在于map函数,它在Python 3中被更改为返回迭代器而不是列表。 Numpy没有正确处理迭代器,所以你需要添加一个list调用来解决问题:

def c1():
    mat=[]
    mat.append(np.array(list(map(float,np.random.rand(4,7)))))
    return mat

您的其余代码有点令人困惑,因为您正在创建一个没有任何有用内容的新字典,然后在不使用索引的情况下枚举新字典的键。看起来您可以使用my_list = [x[0] for x in dict_.values()]并获得相同的结果(因为值列表中的第一项已经是一个庞大的浮点数组,一旦您按上述方法修复c1)。