Python将任意嵌套列表保存到CSV

Question

我有一个由字符串、整数和浮点数组成的列表，以及字符串、整数和浮点数的嵌套列表。这是一个例子

data = [
        1.0,
        'One',
        [1, 'Two'],
        [1, 'Two', ['Three', 4.5]],
        ['One', 2, [3.4, ['Five', 6]]]
    ]

我希望将列表中的每个项目写入 CSV 文件中的一行。因此，根据上述数据，文件将如下所示：

1.0
One
1,Two
1,Two,Three,4.5
One,2,3.4,Five,6

有很多关于如何将列表写入文件的资源，但我还没有看到任何独立列表嵌套的资源。我确信我可以想出一些涉及许多循环等的东西，但是有人有更优雅的解决方案吗？

编辑：我想出的最好的办法是将列表中的每个项目转换为字符串，然后删除多余的字符（“[”、“]”等）。然后附加项目字符串，并将结果写入文件：

string = ''
for i in data:
    line = str(i).replace("[","")
    line = line.replace("]","")
    line = line.replace("'","")
    line = line.replace(" ","")
    string+=line + '\n'

# write string to file...

这只是感觉很笨拙，并且可能有害，因为它假定字符串不包含括号、引号或空格。我正在寻找更好的解决方案！

Answer 1

你问的或多或少是不可能的。

CSV 是一种扁平的表格存储格式。 “任意嵌套列表”的分层性质根本无法很好地匹配或适合表格结构。

您绝对可以展平嵌套列表，以便嵌套列表的每个第一级元素都将出现在输出文件的单行上。但严格来说，这不是 CSV。一些 CSV 阅读器可能会正确读取数据，但其他阅读器不会。而且，一旦按照您的示例进行展平，您将永远无法通过读取文件来重建原始列表。

演示：

[1, ["Two", "Three"], 4.0]

和

[1, ["Two", ["Three"]], 4.0]

两者都会发射：

1
Two,Three
4.0

因此，在读取该文件时，阅读器/解析器将不知道要返回哪个原始列表——第一个二级列表，或第二个三级列表。 （我可以将这个反例任意复杂和丑陋。）

一般来说，嵌套/分层结构和平面/表格结构并不容易或完全兼容。

如果您想要任意嵌套列表的简单存储格式，请考虑JSON 或YAML。它们为嵌套数据提供简单、高质量的存储。例如：

import json

outpath = 'out.json'
with open(outpath, "w") as f:
    f.write(json.dumps(data))

会将您的数据写入文件。重新阅读：

data = json.load(open(out path))

但如果你真的想要 CSV 风格的文本：

def flatten(l):
    """
    Flatten a nested list.
    """
    for i in l:
        if isinstance(i, (list, tuple)):
            for j in flatten(i):
                yield j
        else:
            yield i

def list2csv(l):
    """
    Return CSV-ish text for a nested list.
    """
    lines = []
    for row in l:
        if isinstance(row, (list, tuple)):
            lines.append(",".join(str(i) for i in flatten(row)))
        else:
            lines.append(str(row))
    return "\n".join(lines)

print list2csv(data)

产量：

1.0
One
1,Two
1,Two,Three,4.5
One,2,3.4,Five,6