找到一个匹配的字符串，然后 + 1 到元素

Question

我正在尝试创建一个循环来遍历一个二维列表，如果循环找到第一个元素中的匹配字符串，那么它会将 +1 添加到列表的第二个元素。

列表示例：

[[mike, 0], [john, 2], [henry, 0], [sam, 1]]

如果所需的搜索是约翰，这将发生：

[[mike, 0], [john, 3], [henry, 0], [sam, 1]]

然后是迈克：

[[mike, 1], [john, 3], [henry, 0], [sam, 1]]

在下面的场景中，'name' 是需要找到的字符串。

我还从 CSV 文件中提取名称列表，然后将更新后的列表写回。

我似乎无法让它工作，并且返回给我的错误是

'TypeError: int() 参数必须是字符串、类似字节的对象或数字，而不是'list''

def count(name):
    with open('nameList.csv', 'r') as csvfile:
        reader = csv.reader(csvfile)
        nameList = list(reader)
    for x in range(len(nameList)):
        if nameList[x][0] == name:
            print(nameList== name)
            nameList[1] = nameList[1] + 1
    with open("nameList.csv", "w", newline='') as f:
        writer = csv.writer(f)
        for i in nameList:
            writer.writerow(i)

关于如何使它工作的任何想法？非常感谢！

Answer 1

我相信您要解决的原始问题是获取名称的计数。

如果是这样，那么使用 Python 的 collections 模块中的 Counter 可以更轻松地完成。

from collections import Counter

cnt = Counter()
with open("nameList.csv", 'r') as f:
    rd = csv.reader(f)
    cnt = Counter(r[0] for r in rd)

print(cnt)

cnt 将是类似字典的对象（您可以像使用任何其他字典一样使用它），其中键是 CSV 中的名称，值是计数。所以，它可能看起来像这样：

{'john': 3, 'mike': 1, 'sam': 1}

现在假设您有另一个包含名称的 CSV，并且您想要查找两个文件中每个名称的计数。您可以通过更新cnt 轻松做到这一点：

with open("nameList2.csv", 'r') as f:
    rd2 = csv.reader(f)
    cnt.update(r[0] for r in rd2)

您避免从头开始计数并且代码更性感！简洁大方！

Answer 2

突出的一件事是这一行：

 nameList[1] = nameList[1] + 1

nameList[1] 是主列表中的第二个内部列表。我想你的意思可能是nameList[x][1]：

nameList[x][1] = nameList[x][1] + 1