@li0ard
JS Dev

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

У меня есть вот такой csv файл:
name|id|nick
Test1|1|@test1
Test2|2|@test2
Test3|3|@test3

И как я могу получить все параметры человека по его id в Python т.е есть я ввожу в программу 1 и она выдает имя и никнейм, которые соответствуют id 1
  • Вопрос задан
  • 135 просмотров
Решения вопроса 1
fox_12
@fox_12 Куратор тега Python
Расставляю биты, управляю заряженными частицами
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"]}')
Ответ написан
Комментировать
Пригласить эксперта
Ответы на вопрос 1
@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'}
Ответ написан
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы