如何将嵌套整数列表写入和读取到 csv

Question

我有一个多维矩阵作为 python 模型的数据输入。为此，我使用了保存在不同模型中的嵌套列表，但也使用了 python。我的问题是如何正确地将这些列表保存为 csv 文件 + 如何正确导入它们。

嵌套列表如下所示：

test = [[[3, 16], [10, 12]], [[4, 2], [14, 6]]]

通过阅读其他示例，我尝试使用以下代码保存它们：

# Write CSV file
kwargs = {'newline': ''}
mode = 'w'
if sys.version_info < (3, 0):
    kwargs.pop('newline', None)
    mode = 'wb'

with open('../data/test.csv', mode, **kwargs) as fp:
    writer = csv.writer(fp, delimiter=',')
    writer.writerows(test)

给出如下输出：

"[3, 16]","[10, 12]"
"[4, 2]","[14, 6]"

看着格式，我问自己这是否真的是为更多维度列表编写文件的正确方法。 => Q1：编写嵌套列表的正确方法是什么？

使用读取文件

# Read CSV file
kwargs = {'newline': ''}
mode = 'r'
if sys.version_info < (3, 0):
    kwargs.pop('newline', None)
    mode = 'rb'
with open('../data/test.csv', mode, **kwargs) as fp:
    reader = csv.reader(fp, delimiter=',')
    data_read = [row for row in reader]

有输出

[['[3, 16]', '[10, 12]'], ['[4, 2]', '[14, 6]']]

格式的差异让我想到第二个问题： => Q2：读取文件时如何取回整数？

我相信这个问题不会那么困难，但我刚开始使用 python，所以我不确定如何调整代码。因此，非常感谢您的帮助！

Answer 1

您的问题与 Python 没有任何关系。 CSV 格式不支持嵌套数据结构（例如，列表中的列表）。一种解决方案是使用这样的编码；例如，JSON。或者，如果数据只能由 Python 程序读取，您可以使用 Python 的 pickle 模块来有效地编码/解码对象。

注意：您可以为列表中的列表定义自己的 CSV 元编码。但是你必须想出一个这样做的方案，并为读取序列化数据所需的任何语言编写一个解码器。使用 JSON、protobuf 或其他适当的编码要容易得多。

Answer 2

我之前制作了一个脚本，应该可以解决您的一些问题。

https://github.com/Yolo-gamer/CDM

在我的代码中，我的脚本有两种保存数据的方式。

1. 普通（作为字符串）
2. 数据（以数据类型保存）

只需将 cdm.py 文件与您的脚本放在同一文件夹中，然后使用 import cdm

如需帮助，请键入 help(cdm)，代码应为您提供所需的所有信息。