python字典,列表值为numpy数组

时间:2015-12-13 15:46:45

标签: python arrays numpy scikit-learn

尝试转换具有以下内容的python dict:

  • nteger keys
  • 长度为4的整数列表为值

    {38.0:[139.0,1.8,36.0,18.2]等}

进入一个numpy数组,用sci-kit分析学习贝叶斯岭回归。

1 个答案:

答案 0 :(得分:1)

假设你的意思是我认为你的意思这是一个简单的方法(使用追加时间要短一些,但对于大型数组总是使用numpy预分配)。

    myDict = {38.0: [139.0, 1.8, 36.0, 18.2], 39.0: [139.0, 1.8, 36.0, 18.2]}
    y = np.zeros(len(myDict))
    X = np.zeros((len(myDict), 4))
    i = 0
    for key, values in myDict.iteritems():
        y[i] = key
        X[i, :] = values
        i += 1