python - 将列表分成对角线？

Question

如果我有这个列表列表:

l = [
['a','r','v','s','s','t','n','g'],
['e','x','a','m','p','l','e','t'],
['g','z','n','p','u','t','x','m'],
['v','a','h','g','i','k','t','f']
]

如何根据对角线轻松地将其划分为列表列表？

我想要的结果差不多是:

[
['a'],
['r','e'],
['v','x','g',],
['s','a','z','v'],
['s','m','n','a'],

等等……

我还应该提一下，我已经安装了 numpy 以任何方式提供帮助。

Answer 1

[ [ v[d-i] for i,v in enumerate(l) if d-i>=0 ] for d in xrange(len(l[0])) ]

开发阶段:

[ <one-diagonal-row> for <index d of each element of first row of input> ]

[ [<element-of-diagonal d> for <each v, which is row i of l>] for d in xrange(len(l[0])) ]

[ [ v[d-i] for i,v in enumerate(l) <only if element exists>] for d in xrange(len(l[0])) ]

这会让您到达顶部的导出。