Попытка создать новый фрейм данных из другого фрейма данных, используя информацию, предоставленную из CSV-файла - PullRequest
0 голосов
/ 16 декабря 2018

У меня есть CSV-файл со списком строк: (строки будут меняться каждый день)

-Bob

-Джон

-Frank

У меня есть датафрейм для панд

Заголовок: Имя, возраст, пол

-Shelly 56 Женский

-Jim 48 Мужской

-Frank 50 Мужской

-Джон 52 Мужской

-Боб 53 Мужской

-Мари 40 Женский

Я хотел бы просмотреть каждое имя в файле CSV и получить соответствующую строкуи создайте новый фрейм данных.

1 Ответ

0 голосов
/ 16 декабря 2018
import csv
import pandas as pd

frame = pd.DataFrame({
    'Names': ['A', 'B', 'C', 'D'],
    'Ages': [22, 33, 44, 55],
    'Gender': ['M', 'M', 'F', 'F']
})

names = []

pf = open("test.csv")
csv_reader = csv.reader(pf)

for row in csv_reader:
    names.append(row[0])

new_frame = frame.loc[frame['Names'].isin(names)]

print(new_frame)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...