为什么我的 Python 代码处理 CSV 文件中的标题行而不是跳过它?

Mary-Kate Olsen
发布: 2024-10-30 20:49:02
原创
336 人浏览过

Why is my Python code processing the header row in a CSV file instead of skipping it?

使用 Python 处理 CSV 文件时跳过标头

问题:

在 Python 中,正在处理 CSV 文件,但第一行(标题行)正在被修改而不是被排除。

有问题的代码:

<code class="python">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 processing logic

in_file.close()    
out_file.close()</code>
登录后复制

问题:

将 'row' 变量初始化为 1 不会阻止处理标题行。

解决方案:

要跳过标题行,请使用next() 函数将读取器推进一项可迭代。这种情况下,next()的返回值可以被忽略。

修改代码:

<code class="python">with open("tmob_notcleaned.csv", "rb") as in_file, open("tmob_cleaned.csv", "wb") as out_file:
    reader = csv.reader(in_file)
    next(reader, None)  # Skip the header row
    writer = csv.writer(out_file)

    for row in reader:
        # Row processing logic</code>
登录后复制

替代选项:

如果输出文件中需要标题行,可以在循环之前将其传递给 writer.writerow():

<code class="python">with open("tmob_notcleaned.csv", "rb") as in_file, open("tmob_cleaned.csv", "wb") as out_file:
    reader = csv.reader(in_file)
    headers = next(reader, None)  # Returns the header row or None if the input is empty

    if headers:
        writer.writerow(headers)

    for row in reader:
        # Row processing logic</code>
登录后复制

以上是为什么我的 Python 代码处理 CSV 文件中的标题行而不是跳过它?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板