0

我有一个包含 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']]

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

4

2 回答 2

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文件。

于 2013-10-25T03:20:07.033 回答
1

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

你可以try做这样的事情:

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

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

于 2013-10-25T02:53:03.303 回答