如何将python字典转换为2D数组并在长度不匹配的地方填充?

evan54

我正在尝试将字典项放入数组中。键都是整数,但是每个值都是一个不同长度的列表。我想转换为数组并填充零。我有以下

a = {1:[0, 1], 2:[21], 3:[]}
lmax = 0
for item in a.values():
    lmax = len(item) if len(item) > lmax else lmax
t = np.zeros(len(a), lmax)
for key in a:
    t[ key, :len(a[key]) ] = a[key]

有没有更多的pythonic方式来做到这一点?

谢谢

斯图尔特

您似乎在代码中混用了键和值,所以我不确定该如何工作。

一种方法是

a = {1:[0, 1], 2:[21], 3:[]}
max_length = max(len(item) for item in a.values())
t = [[0] * (max_length - len(item)) + item for item in a.values()]

或者,如果您想确保列表按“正确”的顺序排列(即,原始字典中具有最低键的项首先显示),则可以将其更改为

t = [[0] * (max_length - len(item)) + item for _, item in sorted(a.items())]

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章