在 python 3.3 中导入 csv 作为列表列表

Question

我有一个包含 2 个字符串和 1 个整数的列表

list_of_ee = [["a","m",15],["w","p",34]]

我使用此代码将其导出到 csv 文件。

import csv

myfile = open("pppp.csv", 'wb')
with open("pppp.csv", "w",newline='') as myfile:
    wr = csv.writer(myfile, quoting=csv.QUOTE_NONE)
    wr.writerows(list_of_ee)

结果是

a m 15
w p 34

我使用此代码导出它

data = csv.reader(open('pppp.csv','r', newline=''))
data = list(data)
print(data)

结果是

[['a', 'm', '15'], ['w', 'p', '34']]

但我想在list_of_ee 中导入与之前相同的结果（作为列表列表和 2 个 str 和 1 个整数）（当我的程序重新启动时） 我想将它们添加到list_of_ee（因为程序启动时没有数据 list_of_ee.

Answer 1

由于 CSV 格式没有数据类型的概念，因此没有 CSV 内置的方式来执行您所要求的操作；也就是说，它不区分字符串和整数。

你可以 try 做这样的事情：

for lst in list_of_ee:
    for i,l in enumerate(lst):
        try:
            lst[i] = int(l)
        except:
            pass

将子列表的数字元素转换为整数。

Answer 2

如果你使用 CSV，你会得到字符串。请参阅文档：http://docs.python.org/2/library/csv.html#module-contents

“不执行自动数据类型转换。”

@hcwhsa 为您指出了一个答案，该答案显示了如何使用ast.literal_eval() 来猜测类型。难度不大：

import ast
import csv

def convert_type(s):
    try:
        return ast.literal_eval(s)
    except (ValueError, SyntaxError):
        return s

def convert_csv_row(lst):
    return [convert_type(x) for x in lst]

data = csv.reader(open('pppp.csv','r', newline=''))
converted = [convert_csv_row(row) for row in data]
print(converted)

但实际上，如果您想保留类型，为什么还要使用 CSV？除非您将数据导出到电子表格或其他东西，否则我建议您使用 JSON 格式。

import json

list_of_ee = [["a","m",15],["w","p",34]]

with open("test.json", "wt") as f:
    f.write(json.dumps(list_of_ee))

with open("test.json", "rt") as f:
    s = f.read()
    lst = json.loads(s)

print(lst)
assert list_of_ee == lst

JSON 是将数据导出到其他程序的好方法。

但如果您只是想为自己的 Python 程序保存数据，那就更简单了：只需使用 pickle。 http://docs.python.org/2/library/pickle.html

import pickle

list_of_ee = [["a","m",15],["w","p",34]]

with open("test.pickle", "wb") as f:
    f.write(pickle.dumps(list_of_ee))

with open("test.pickle", "rb") as f:
    bytes_data = f.read()
    lst = pickle.loads(bytes_data)

print(lst)
assert list_of_ee == lst

使用pickle，您需要以二进制模式写入和读取文件，而不是文本模式。同样使用pickle，您可以保存几乎任何 Python 原生类型，而不仅仅是 JSON 支持的基本类型。但几乎只有 Python 程序会读取 pickle 文件。