python - 给定行号和列号的情况下如何更新csv值

Question

我有一些代码可以查看 .csv 文件的行并检查这些行是否包含来自另一个 .csv 文件的特定列的任何值。

这些文件如下所示：

查找文件：

0:  TextExclude, text, other
1:  aa,        , x ,   y
2:  bb,        , x ,   y
3:  cc,        , x ,   y

我要在其中找到这些值的文件：

0: x, longtext, exclude
1: x, helloaa,  0
2: x, testaa,   0
3: x, testcc,   0
4: x, no,       0
5: x, aabb,     0

我的代码的输出应该在除第 4 行之外的每一行中将“包含”列的值从 0 更改为 1，从而产生此 预期输出 csv 表：

0: x, longtext, exclude
1: x, helloaa,  1
2: x, testaa,   1
3: x, testcc,   1
4: x, no,       0
5: x, aabb,     1

由于我的代码可以输出找到匹配项的 行号 并且 列号 已经定义，我想知道解决此问题的最佳方法是什么并相应地更新 .csv 文件？

这是我的代码：

import pandas
findlist = []
linecount=0
       
with open('lookup.csv', 'r') as f:
    column_names = ["TextExclude", "Exclusion", "Filename"]
    r = pandas.read_csv(f, names=column_names)
    findlist = r.TextExclude.to_list()

with open('datafile.csv', 'r') as f:
    # Skip the first line
    f.readline()
    for line in f: 
        linecount = linecount +1
        if any(listelement in line for listelement in findlist):
            print(line)

Answer 1

您应该使用 pandas 解析 datafile.csv，而不仅仅是将其读取为纯文本文件。这样一来，您就可以将搜索隔离到正确的列，并更轻松地更新第三列。

import pandas
findlist = []
linecount=0

with open('lookup.csv', 'r') as f:
    column_names = ["TextExclude", "Exclusion", "Filename"]
    r = pandas.read_csv(f, names=column_names)
    findlist = r.TextExclude.to_list()

with open('datafile.csv', 'r') as f:
    df = pandas.read_csv(f)
    for ri, row in df.iterrows():
        if any(x in row[1] for x in findlist):
            row[2] = "1"

# print result to stdout
print(df)

# or write result to a file:
df.to_csv("output.csv")

如果您想知道您使用的是什么版本的 Pandas 和 Python，您可以运行以下两个命令：

Python 版本：
python --version

熊猫版：
python -c "import pandas;print(pandas.__version__)"

这是第二种搜索方式，它可以将步骤分解得更细一些，这可能有助于调试：

import pandas
findlist = []
linecount=0

with open('lookup.csv', 'r') as f:
    column_names = ["TextExclude", "Exclusion", "Filename"]
    r = pandas.read_csv(f, names=column_names)
    findlist = r.TextExclude.to_list()

def search(search_in):
    for to_find in findlist:
        if to_find in search_in:
            return True
    return False

with open('datafile.csv', 'r') as f:
    df = pandas.read_csv(f)
    for ri, row in df.iterrows():
        if search(row[1]):
            row[2] = "1"

# print result to stdout
print(df)

# or write result to a file:
df.to_csv("output.csv")