当前位置:嗨网首页>书籍在线阅读

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 中。