尝试使用csv文件提供的信息从另一个数据框创建新的数据框

时间:2018-12-15 21:59:09

标签: python pandas

我有一个带有字符串列表的csv文件:(字符串每天都会更改)

-鲍勃

-约翰

-坦率

我有一个熊猫数据框

标题:姓名年龄性别

-雪莉56女

-Jim 48男

-弗兰克50男

-约翰52男

-鲍勃53男

-玛丽40岁女性

我想遍历csv文件中的每个名称并检索匹配的行并创建一个新的数据框。

1 个答案:

答案 0 :(得分:0)

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)