将字符串列表（csv）转换为浮点列表

Question

我有一些困难将字符串列表转换为浮点列表。我尝试了这两种方法，每个都返回不同的错误。

import csv 
import math 

unemp_reader = csv.reader(open('unemp.csv', 'rU')) 
unemp_lines = list(unemp_reader) 

for rows in unemp_lines:  #tried this way, but error tells me indices must be integers 
    i = 1 
    for i in rows: 
     a = map(float, unemp_lines[i]) 
     float_list.append(a) 
    print float_list 

for row in unemp_lines: #tried this way but the list returned is empty 
    y = row[1].split(",")[1:-1] 
    float_list = [float(i) for i in y if i] 
print float_list 

Answer 1

在您的第一个示例中，您的问题是for循环没有给出索引，它给出了列表中的值。这意味着你的例子根本没有任何意义。

第二个示例将该行中的第二个项目分割为逗号（无论如何都应该由csv模块处理），然后将第二个项目作为结果列表的最后一项。正如我想象的那样，在这个值中没有逗号，这将带有一个元素的列表[1:-1]，返回一个空列表。我无法真正理解这里的意图。您也只能存储最后一行的数据（每次覆盖float_list）。你似乎是在猜测csv模块，并且使它比现在困难得多。

你需要停下过于复杂吧：

with open('unemp.csv', 'rb') as data: 
    rows = csv.reader(data) 
    next(rows) #Skip the headers. 
    floats = [[float(item) for number, item in enumerate(row) if item and (1 <= number <= 12)] for row in rows] 

print(floats) 

为了解释，首先我们使用with语句来打开的可读性和Python的方式文件（确保文件被正确关闭，甚至在例外） 。然后，我们制作一个csv.reader以从CSV文件获取我们的数据。我们通过将迭代器前进1来跳过标题，这意味着我们从第二行开始。然后，我们使用列表理解从迭代器生成一个新列表，其中包含另一个列表理解，如果这些值存在，则生成值的浮点数，但不在Year或Annual列中。为此，我们使用内建的enumerate()来获得我们所在列的编号，然后进行检查以确保它不是0（Year）或13（Annual）。

由于J.F.Sebastian在评论中指出，最好的解决办法就是让csv模块来处理与数字打交道的你，通过与价值csv.QUOTE_NONNUMERIC加上命名参数quoting到csv.reader()通话。例如：

with open('unemp.csv', 'rb') as data: 
    rows = csv.reader(data, quoting=csv.QUOTE_NONNUMERIC) 
    next(rows) #Skip the headers. 
    floats = [[item for number, item in enumerate(row) if item and (1 <= number <= 12)] for row in rows] 

print(floats)