How to find string in CSV file python-3

Question

I have a program that takes file code as input in this case '19E071OE1' and it needs to print out all the rows that have that value in its 4th column it kind of works but I have to input file code for each individual row. My question is how do I make it scan all rows at once and print them out?

Code:

import csv


def ime_predmeta():
    imepred = input("Unesi kod predmeta: ")
    if imepred.isupper():
        return imepred
    else:
        print("Unos nije validan probajete opet!")
        return ime_predmeta()


def ucitavanjerasp1():
    file = open('raspored1.csv', "r")
    reader = csv.reader(file, delimiter=',')
    for row in reader:
        if ime_predmeta() in row[3]:
            print(row)


def main():
    ucitavanjerasp1()


if __name__ == '__main__':
    main()

The bad output:

Unesi kod predmeta: 19E071OE1
['0', '08:00', '11:00', '19E071OE1 [OO 2019] P1']
Unesi kod predmeta: 19E071OE1
['0', '11:00', '14:00', '19E071OE1 [OO 2019] P2']
Unesi kod predmeta: 19E071OE1
['0', '14:00', '17:00', '19E071OE1 [OO 2019] P3']
Unesi kod predmeta: 19E071OE1

Leo Arad · Accepted Answer · 2021-01-12 13:00:12Z

1

You can achieve that by using regex split method like

with open('raspored1.csv', "r") as f:
    content = f.read()
    print(re.split(f'.*\,.*\,.*\,.*{imepred}.*',content))

this will return you a list with all lines that contain imepred value init in the 4 column.

edited Jan 12, 2021 at 13:00

answered Jan 12, 2021 at 12:54

Leo Arad

4,4822 gold badges8 silver badges17 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to find string in CSV file python-3

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related