python：将一个表中的数据和另一个表中相同的相关数据提取出来

weixin_43384009

于 2019-03-12 10:19:56 发布

阅读量1w

点赞数 7

分类专栏： python 文章标签： pandas 数据提取

本文链接：https://blog.youkuaiyun.com/weixin_43384009/article/details/88415515

版权

python 专栏收录该内容

62 篇文章

订阅专栏

本文介绍了一种使用Python的pandas库从两个CSV文件中读取数据，并通过内连接方式将它们合并的方法。首先，从指定路径读取两个CSV文件，然后利用pandas将这些数据转换为DataFrame。接下来，通过内连接的方式，基于'id'、'trip'和'time'字段将两个DataFrame进行合并。最后，将合并后的数据保存到新的CSV文件中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import pandas as pd
import csv
from pandas.core.frame import DataFrame

trip_info=open("D:/task/data/60/60trip_test1.csv")    #打开表一
radar_info=open("D:/task/data/60/radar_test1.csv")  #打开表二

trip_reader=csv.reader(trip_info)  #逐行读取表一数据
list1=[]
n=0
for onea_line in trip_reader:
    list1.append(onea_line)
    n+=1
#print(number)
#print(data1[1][2])

radar_reader=csv.reader(radar_info)  #逐行读取表二数据
list2=[]
m=0
for oneb_line in radar_reader:
    list2.append(oneb_line)
    m+=1
#print(m)
#print(data2[1][0])

#按照相同的数据提取另一个表中的相关信息
df1=pd.DataFrame(list2,columns['id','trip','time','target','longitudinal_radar','broadwise_radar','longitudinal_speed','broadwise_speed','target_vehicle','target_move','a'])
#print(df1)
df2=pd.DataFrame(list1,columns=['id','trip','time'])
#print(df2)
df=pd.merge(df1,df2,how='inner',on=['id','trip','time'],right_index=True)
df.sort_index(inplace=True)
#print(df)

#将dataframe写入csv文件
df.to_csv("choice_radar.csv")
read_csv=pd.read_csv("choice_radar.csv")
print(read_csv)