10-第2步_读入CSV文件
第2步:读入CSV文件
该程序不会从原来的CSV文件删除第一行。但是,它会创建新的CSV文件副本,不包含第一行。因为副本的文件名与原来的文件名一样,所以副本会覆盖原来的文件。
该程序需要有一种方法,用于了解它的循环当前是否在处理第一行。为removeCsvHeader.py添加以下代码:
#! python3
# removeCsvHeader.py - Removes the header from all CSV files in the current
# working directory.
--snip--
# Read the CSV file in (skipping first row).
csvRows = []
csvFileObj = open(csvFilename)
readerObj = csv.reader(csvFileObj)
for row in readerObj:
if readerObj.line_num == 1:
continue # skip first row
csvRows.append(row)
csvFileObj.close()
# TODO: Write out the CSV file.
reader
对象的 line_num
属性可以用来确定当前读入的是CSV文件的哪一行。另一 个for
循环会遍历CSV reader
对象并返回所有行。除了第一行,所有行都被添加到 csvRows
中。
在 for
循环遍历每一行时,代码检查 readerObj.line_num
是否设为 1
。如果是这样,它执行 continue
转向下一行,且不将它添加到 csvRows
中。对于之后的每一行,条件永远是 False
,这些行将添加到 csvRows
中。