Python循环遍历CSV文件及其列

Question

所以我已经看到这里完成了其他问题，但我仍然有点困惑。过去几天我一直在学习 python3，我想我会开始做一个项目来真正弄脏我的手。我需要遍历一定数量的 CSV 文件并对这些文件进行编辑。通常，我无法转到特定列以及 python 中的循环。我习惯了约定（int i = 0; i

import os
import csv

pathName = os.getcwd()

numFiles = []
fileNames = os.listdir(pathName)
for fileNames in fileNames:
    if fileNames.endswith(".csv"):
        numFiles.append(fileNames)

for i in numFiles:
    file = open(os.path.join(pathName, i), "rU")
    reader = csv.reader(file, delimiter=',')
    for column in reader:
        print(column[4])

我的问题在于这一行：

for column in reader:
        print(column[4])

所以在文档中它说 column 是变量，而 reader 是我正在循环的内容。但是当我写 4 我得到这个错误：

IndexError: list index out of range

这是什么意思？如果我写 0 而不是 4，它会打印出每个 CSV 文件的第 0 列单元格 0 中的所有值。我基本上需要它遍历每个 CSV 文件的第一行并找到一个特定的值，然后遍历整个列。提前致谢！

Answer 1

可能是您的 .csv 文件中没有 5 列。

Python 是 base0，这意味着它从 0 开始计数，因此第一列为 column[0]，第二列为 column[1]。

你也可能想改变你的

for column in reader:

到

for row in reader:

因为 reader 遍历的是行，而不是列。

此代码循环遍历每一行，然后是该行中的每一列，以便您查看每个单元格的内容。

for i in numFiles:
    file = open(os.path.join(pathName, i), "rU")
    reader = csv.reader(file, delimiter=',')
    for row in reader:
        for column in row:
            print(column)
            if column=="SPECIFIC VALUE":
                #do stuff

Answer 2

欢迎来到 Python！我建议你打印一些调试信息。

您可以将其添加到您的打印循环中：

for row in reader:
    try:
        print(row[4])
    except IndexError as ex:
        print("ERROR: %s in file %s doesn't contain 5 colums" % (row, i))

这将打印坏行（作为列表，因为这是它们在 CSVReader 中的表示方式），因此您可以修复 CSV 文件。

一些注意事项：

1. 在 Python 中通常使用 snake_case 而不是 camelCase
2. 适当地命名您的变量（csv_filename 而不是 i、row 而不是 column 等）
3. 使用with close 处理文件 (read more)

享受吧！