Как получить данные по 1 параметру из CSV?

Question

[Artem Petrenko]

У меня есть вот такой csv файл:

name|id|nick
Test1|1|@test1
Test2|2|@test2
Test3|3|@test3

И как я могу получить все параметры человека по его id в Python т.е есть я ввожу в программу 1 и она выдает имя и никнейм, которые соответствуют id 1

Answer 1

[Владимир Куц]

import csv

ID = 1

with open('users.csv') as csvfile:
    reader = csv.DictReader(csvfile, delimiter='|')
    for row in reader:
        if int(row['id']) == ID:
            print(f'{row["name"]} {row["nick"]}')

Answer 2

[zexer]

Можно сделать так

import pandas as pd

data = pd.read_csv('list.csv', sep=';')


def get_user_by_id(id, data):
    data = data.loc[data['id'] == id, ['nick', 'name']]
    name = data['name'][0]
    nick = data['nick'][0]
    return {'name': name, 'nick': nick}


print(get_user_by_id(1, data))
# Out: {'name': 'Test1', 'nick': '@test1'}

Comments

[Artem Petrenko]

Не работает

pandas.errors.ParserError: Error tokenizing data. C error: Expected 7 fields in line 333, saw 8

[zexer]

Николай Коновалов, вы правильно прочитали файл? Вместо ';' подставили свое значение сепаратора?
https://ru.stackoverflow.com/questions/815001/%D0%...