我正在尝试读取一个csv文件,并根据其中存储的值创建一个二维列表。
但是,当我尝试检查存储的值是否可以转换为浮点数时,我遇到了麻烦。
这是我编写的函数,该函数读取文件并创建列表。
def readfile(amount, name):
tempfile = open(name).readlines()[1:] #First value in line is never a float, hence the [1:]
rain_list = []
count = 0.0
for line in tempfile:
line = line.rstrip()
part = line.split(",")
try:
part = float(part)
except ValueError:
print("ERROR: invalid float in line: {}".format(line))
rain_list.append(part[amount])
count += 1
if count == 0:
print("ERROR in reading the file.")
tempfile.close()
return rain_list
这可能有点混乱,因为它本质上是我尝试过的各种可能解决方案的拼凑而成。
它获取的值是文件的名称(名称)和从文件中读取的值的数量(数量)。
有人知道为什么它不能按我预期的那样工作吗?
答案 0 :(得分:2)
part
是一个字符串列表。要检查并转换所有浮动,您需要执行以下操作:
part = [float(x) for x in part]
(包装在您的异常块中)
顺便说一句,您应该使用csv
模块来读取逗号分隔的文件。它是内置的。同样使用enumerate
可以打印错误发生的行,而不仅仅是数据:
reader = csv.reader(tempfile) # better: pass directly the file handle
# and use next(reader) to discard the title line
for lineno,line in enumerate(reader,2): # lineno starts at 2 because of title line
try:
line = [float(x) for x in line]
except ValueError:
print("ERROR: invalid float in line {}: {}".format(lineno,line))