Поиск и вывод на экран с csv файла
Есть csv файл с списком студентов и их оценкой,нужно ввести ,допустим, и чтобы вывело его номер строки в файле и оценку,как это можно реализовать?
name,mark
A,100
Igor,74
Anton,75
Victor,88
B,4
V,5
Ответы (1 шт):
Автор решения: MaxU
→ Ссылка
парсинг CSV:
import pandas as pd
import numpy as np
df = pd.read_csv(filename)
# start indexing from 1
df.index = np.arange(1, len(df) + 1)
поиск и вывод на экран:
In [8]: print(df.query("name == 'Anton'"))
Out[8]:
name mark
3 Anton 75
PS значения индекса соответствуют номеру строки в CSV файле
PPS данное решение не будет правильно работать если в ячейках CSV файла присутствуют символы перевода строки "\n".
UPDATE: поиск при помощи переменных:
In [10]: find = 'Anton'
In [11]: print(df.query("name == @find"))
name mark
3 Anton 75
In [12]: names = ["Anton", "B"]
In [13]: print(df.query("name in @names"))
name mark
3 Anton 75
5 B 4