扩展列表列表

Question

我正在寻找一种方法来优化我拥有的当前代码：

        for result in results:
            if result != '':
                if isinstance(result, list):
                    for subresult in result:
                        if subresult != '':
                            worksheet.write(r, c, subresult)
                            c += 1
                else:
                    worksheet.write(r, c, result)
                    c += 1

我也在特殊情况下使用这个：

        for result in results:
            if isinstance(result, list):
                for subresult in result:
                    worksheet.write(r, c, subresult)
                    c += 1
            else:
                worksheet.write(r, c, result)
                c += 1

我有一个列表，在某些情况下只包含一个值，而在其他情况下它包含多个值（即另一个列表）。

以下是列表可能包含的示例：

results = ['', '', '', '390', '66', ['Name', 'SerialNumber', 'Model', 'Year'], 'SW_Version', ['HD_Loc', 'HD_Model', 'HD_FW', 'HD_SerialNumber', 'Man_Yr', 'Man_Mth'], '', '', '']

最终我希望能够将每个值输出到 excel 中自己的列。在某些情况下，我只想输出列表中包含值的部分，这是由我提供的代码的第一个 sn-p 执行的。在其他情况下，我想输出列表的所有部分，无论它是否有值，这是由我提供的代码的第二个 sn-p 执行的。

我已经看到 for 循环和一些操作语句（比如说 print）被组合成一行 python 代码的实例，但我无法让该语法与 sn- ps 我已经提供了。

此代码目前正在以我实现它的方式工作，我只是在寻找一种缩短/优化它的方法，因为我拥有它的程序开始变得相当大并且能够缩短它会有所帮助我也在缩短程序的其他部分。

提前致谢！

编辑：在这两种情况下，它的输出顺序都很重要，我希望它按照当前的顺序输出（只需减去 '' 中的一种情况）。

Answer 1

希望对你有帮助，

results = ['', '', '', '390', '66', ['Name', 'SerialNumber', 'Model', 'Year'], 'SW_Version', ['HD_Loc', 'HD_Model', 'HD_FW', 'HD_SerialNumber', 'Man_Yr', 'Man_Mth'], '', '', '']

flatList = [item for sublist in [[item] if not isinstance(item,list) else item for item in results] for item in sublist]

flatListWithoutBlanks = [item for item in [item for sublist in [[item] if not isinstance(item,list) else item for item in results] for item in sublist] if item != '' ]

最好的，

Answer 2

好的，这就是我目前得到的解决方案。这是一个有点混乱的列表理解。我会试着清理一下。

results = ['', '', '', '390', '66', ['Name', 'SerialNumber', 'Model', 'Year'], 'SW_Version', ['HD_Loc', 'HD_Model', 'HD_FW', 'HD_SerialNumber', 'Man_Yr', 'Man_Mth'], '', '', '']
li = list(set([item if type(sublist) is list else sublist for sublist in results for item in sublist]))
print li

这给出了以下输出：

['SW_Version', 'Name', 'HD_FW', 'SerialNumber', 'HD_Model', 'HD_SerialNumber', '390', '66', 'Year', 'Model', 'Man_Yr', 'HD_Loc', 'Man_Mth']

Answer 3

你可以用递归巧妙地做到这一点：

def worksheet_writer(results):
        if isinstance(results, list):
                for item in results: worksheet_writer(item)
        else:
                worksheet.write(r, c, results)
                c += 1

已编辑以支持字符串。

Answer 4

我在评论中发布了它，但不确定它是否清楚，

flatListWithoutBlanks = [item for item in [item for sublist in [[item] if not isinstance(item,list) else item for item in results] for item in sublist] if item != '' ]