我有一个数据框架,它有一万行。 我需要根据一个条件(其中name在name_list中)将行提取到一个新的数据框架中
做这件事最简单的方法是什么
您可以使用for循环遍历df中的每一行
for index,row in df.iterrows():
print(row)//row is an array which contain your each column data'[name, age,....]
因此您可以在for循环中使用您的条件并筛选行
也可以通过调用df.apply(UDF())使用UDF函数来完成此操作
假设这是您的数据框架结构和名称:dataframe=pd.dataframe(record,columns=['name','age','stream','percentic']
因此,新数据帧可以创建为:
rslt_df=dataframe[dataframe['name'].isin(name_list)]
或者,您可以使用:
rslt_df=dataframe.loc[dataframe['name'].isin(name_list)]