我有一个 Python 3 字典,其中包含很长的列表(每个列表有 3000 万个整数)。我想将所有这些列表拼接成一个 numpy 数组。我怎样才能高效做到这一点?
以下内容
np.array(my_dict.values())
似乎不起作用(我得到的是 array(dict_values([[...], [....]))
而不是平面 1D numpy 数组)。
最佳答案
如果您正在寻找平面一维数组,您可以使用 np.concatenate
:
>>> d = {'a': [1, 2, 3, 4, 5], 'b': [1, 2, 3, 4, 5], 'c': [1, 2, 3, 4, 5]}
>>> np.concatenate(list(d.values()))
array([1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5])
关于python - 将列表字典转换为一维 numpy 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25942293/