Skip to main content
 首页 » 编程设计

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

2025年05月04日10kenshinobiy

在 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填充值,但必须有更好的方法.

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

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

请您参考如下方法:

您可以创建具有所需尺寸的 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')