读取 csv 文件以列出

Question

我想导出一个嵌套列表，然后在不丢失列表形状的情况下导入它。该列表是一个嵌套列表，每个内部列表有10个元素，有10000个内部列表构成一个大列表。

内部列表如下所示：形状为 1 x 10

    [array([  1, 2, 3, 4, 5

              6, 7, 8, 9, 10 ],

    array([ 21, 22, 23, 24, 25

           26, 27, 28, 29, 30 ]

我将它导出到csv。在csv中，它变成了这样。每个内部列表都在一个单元格中

['[ 1  2  3  4  5\n 6  7  8  9  10]']

['[ 21  22  23  24  25\n 26  27  28  29  30]']

然后我将 csv 导入到列表中。但是形状变成 1 x 1 我希望将其变成 1 x 10 形状，理想情况下与我导出的原始列表格式相同。如何实现这一目标？

我用于导出列表的代码：（X 是列表）

import csv

with open('X.csv', "w") as output:

    writer = csv.writer(output, lineterminator='\n')

    for val in X:

        writer.writerow([val])  

我用于导入csv的代码：（X是列表）

with open('X.csv', 'r') as f:

    reader = csv.reader(f)

    X = list(reader)

先谢谢了！

Answer 1

我建议你可以将它保存为泡菜或数据帧，而不是在读回它时你不需要摆弄这么多。泡菜保存在 python 使用的内存表示中。import picklepickle.dump(X, open('X.pkl', "wb"))X_loaded = pickle.load(open('X.pkl', "rb"))