首页 > 解决方案 > 如何在列中搜索变量,然后使用 Python 3 打印出包含 csv 文件中数字的所有行

问题描述

我正在制作一个程序来检查 CSV 中的数字,然后打印出整行。第四个是数字。这是我尝试过的:

 with open('Info.csv', 'r') as csv_file
 if 'View C' in choice:
        read = csv.reader(csv_file)
        view = str(input("Enter Number: "))
        for column in read:
            if view == column[3]:
                print(row)

csv 文件的结构如下:

John,Smith,London,131390890
Bob,Builder,Moscow,123123132
Dab,God,LA,131390890

我希望程序是这样的:

输入:

123123132

输出:

Bob,Builder,Moscow,123123132,

它还需要能够做到这一点:

输入:

131390890

输出:

John,Smith,London,131390890,
Dab,God,LA,131390890,

谢谢!

顺便说一句,使用 Python 3 ...

标签: pythonpython-3.xcsv

解决方案


您可以使用列表理解。要查找具有您要查找的数字的行,请阅读整个文件,将每一行附加到名为content. 然后:

result = [i for i in content if i[-1] == 'your_number']

这样,变量result将包含 N 行,每行的最后一列与您的数字匹配


推荐阅读