在Python中使用CSV模块查找包含输入字符串的行

时间:2017-05-12 02:10:58

标签: python-2.7 csv

我正在尝试获取与我输入的“profile_name”匹配的行的所有内容 这是CSV布局:

    profile_name,first_name,last_name,age

这是信息:

    Johnny,Johnny,Appleseed,25

我尝试过很多东西,但我不懂CSV模块。 任何澄清/帮助都非常感谢

1 个答案:

答案 0 :(得分:0)

csv模块允许您将文件中的每一行作为列列表。

然后,您可以在预期的列中搜索该值。

这些方面的东西:

import csv

# wait for user to input the profile to search
profile_to_search = input()

with open('profiles.csv', 'rt') as f:
    reader = csv.reader(f, delimiter=',')
    for row in reader:
        if profile_to_search == row[0]:
            print row