用Python编辑csv文件时跳过标题

2024-04-20 08:10:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用下面提到的代码使用Python编辑csv。在代码中调用的函数构成代码的上部。

问题:我希望下面引用的代码从第二行开始编辑csv,我希望它排除包含标题的第一行。现在它只在第一行应用函数,而我的标题行正在被更改。

in_file = open("tmob_notcleaned.csv", "rb")
reader = csv.reader(in_file)
out_file = open("tmob_cleaned.csv", "wb")
writer = csv.writer(out_file)
row = 1
for row in reader:
    row[13] = handle_color(row[10])[1].replace(" - ","").strip()
    row[10] = handle_color(row[10])[0].replace("-","").replace("(","").replace(")","").strip()
    row[14] = handle_gb(row[10])[1].replace("-","").replace(" ","").replace("GB","").strip()
    row[10] = handle_gb(row[10])[0].strip()
    row[9] = handle_oem(row[10])[1].replace("Blackberry","RIM").replace("TMobile","T-Mobile").strip()
    row[15] = handle_addon(row[10])[1].strip()
    row[10] = handle_addon(row[10])[0].replace(" by","").replace("FREE","").strip()
    writer.writerow(row)
in_file.close()    
out_file.close()

我试图通过将row变量初始化为1来解决这个问题,但它没有工作。

请帮助我解决这个问题。


Tags: csv函数代码in编辑标题openout
3条回答

解决这个问题的另一种方法是使用DictReader类,该类“跳过”头行并使用它来允许命名索引。

给出“foo.csv”如下:

FirstColumn,SecondColumn
asdf,1234
qwer,5678

像这样使用听写器:

import csv
with open('foo.csv') as f:
    reader = csv.DictReader(f, delimiter=',')
    for row in reader:
        print(row['FirstColumn'])  # Access by column header instead of column number
        print(row['SecondColumn'])

执行row=1不会改变任何内容,因为您只需用循环的结果覆盖它。

要跳过一行,请执行next(reader)

您的reader变量是一个iterable,通过在其上循环您可以检索行。

要使它在循环之前跳过一项,只需调用^{}并忽略返回值。

您还可以稍微简化代码;将打开的文件用作上下文管理器以自动关闭它们:

with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
   reader = csv.reader(infile)
   next(reader, None)  # skip the headers
   writer = csv.writer(outfile)
   for row in reader:
       # process each row
       writer.writerow(row)

# no need to close, the files are closed automatically when you get to this point.

如果要将头写入未处理的输出文件,也很容易,请将next()的输出传递给writer.writerow()

headers = next(reader, None)  # returns the headers or `None` if the input is empty
if headers:
    writer.writerow(headers)

相关问题 更多 >