从python列表中提取数据

Question

我正在使用一个以[[11,3.22），（12,4.6），（9,2.4）]格式显示结果的程序，我需要提取第一部分以将其更改为名称，然后附加第二个值并将其存储在CSV文件。 如何提取子部分的每个部分？

Answer 1

>>> l = [(11,3.22),(12,4.6),(9,2.4)]

“需要提取第一部分”-

>>> l[0]
(11, 3.22)

“将其更改为名称”-

>>> l[0] = ('x', 'y')
>>> l
[('x', 'y'), (12, 4.6), (9, 2.4)]

“附加第二个值”-

>>> l[0] = dict(zip(('x', 'y'), l[1]))
>>> l
[{'y': 4.6, 'x': 12}, (12, 4.6), (9, 2.4)]

以CSV格式存储很容易，请在此处查看示例-http: //www.pythonforbeginners.com/systems-programming/using-the-csv-module-in-python/

Answer 2

我假设您的意思是对于列表中的每个元组，将第一个元素替换为从整数映射的字符串。

您可以使用列表推导来执行此操作：

>>> id_to_str_map = {11:"foo", 12:"bar", 9:"baz"}
>>> l = [(11,3.22),(12,4.6),(9,2.4)]
>>> result = [(id_to_str_map[idx], value) for (idx, value) in l]
>>> print result
[('foo', 3.22), ('bar', 4.6), ('baz', 2.4)]

使用@theharshest建议的CSV标准库模块是最可靠的选择。 适用于Python 2.7的标准库文档： http : //docs.python.org/2/library/csv.html

如果您正在使用大型数据集，那么最好在将每一行写到csv文件中时，使用生成器表达式来懒惰地执行映射。

import csv
id_to_str_map = {11:"foo", 12:"bar", 9:"baz"}
l = [(11,3.22),(12,4.6),(9,2.4)]
with open("blah.csv", "wb") as csvfile:
    csv_writer = csv.writer(csvfile)
    for row in ((d[idx], value) for (idx, value) in l):
        csv_writer.writerow(row)