python - 将不完整的嵌套列表放入矩形 ndarray

Question

在 Python 中(也使用 numpy)我有一个列表的列表列表，每个列表的长度不同。

[
    [
         ["header1","header2"],
         ["---"],
         [],
         ["item1","value1"]
    ],

    [
         ["header1","header2","header3"],
         ["item2","value2"],
         ["item3","value3","value4","value5"]
    ]
]

我想使这个数据结构成为矩形:即保证 len(list[x]) 对于所有 x 都是常量，len(list[x ][y]) 对于所有 x、y 等都是常量。

(这是因为我要将数据结构导入到numpy中)

我可以想到各种非Python的方式来做这样的事情(遍历结构，记录每个级别的最大长度，第二次通过并使用None填充值，但必须有更好的方法.

(我也希望解决方案不依赖于结构的维度；即它也应该适用于此类结构的列表...)

有没有我缺少的简单方法？

Answer 1

您可以创建具有所需尺寸的 ndarray 并轻松读取您的列表。由于您的列表不完整，您必须捕获 IndexError，这可以在 try/exception block 中完成。

使用 numpy.ndenumerate 可以轻松地将解决方案扩展到更多维度(添加更多索引 i,j,k,l,m,n,...在下面的 for 循环中):

import numpy as np
test = [ [ ["header1","header2"],
           ["---"],
           [],
           ["item1","value1"] ],
         [ ["header1","header2","header3"],
           ["item2","value2"],
           ["item3","value3","value4","value5"] ] ]


collector = np.empty((2,4,4),dtype='|S20')

for (i,j,k), v in np.ndenumerate( collector ):
    try:
        collector[i,j,k] = test[i][j][k]
    except IndexError:
        collector[i,j,k] = ''


print collector
#array([[['header1', 'header2', '', ''],
#        ['---', '', '', ''],
#        ['', '', '', ''],
#        ['item1', 'value1', '', '']],
#       [['header1', 'header2', 'header3', ''],
#        ['item2', 'value2', '', ''],
#        ['item3', 'value3', 'value4', 'value5'],
#        ['', '', '', '']]],  dtype='|S10')