我有一个带有字符串列表的csv文件:(字符串每天都会更改)
-鲍勃
-约翰
-坦率
我有一个熊猫数据框
标题:姓名年龄性别
-雪莉56女
-Jim 48男
-弗兰克50男
-约翰52男
-鲍勃53男
-玛丽40岁女性
我想遍历csv文件中的每个名称并检索匹配的行并创建一个新的数据框。
答案 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)