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

Question

我有一个带有字符串列表的 csv 文件：（字符串每天都会改变）

-鲍勃

-约翰

-坦率

我有一个熊猫数据框

标题：姓名年龄性别

-雪莉 56 女

- 吉姆 48 男

- 弗兰克 50 男

-约翰 52 男

-鲍勃 53 男

-3 月 40 日女性

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

标签： pythonpandas

Answer 1

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)