课程设计-在校整理-09 基于轨迹数据的传染结果挖掘

【在校整理-09 基于轨迹数据的传染结果挖掘】(注:仅供参考学习使用)

一、课题内容和要求

1、问题描述

移动通信技术及全球定位和导航系统等定位技术的的迅速发展,由此产生了大量的用户行程轨迹数据,并造成大量轨迹数据的堆积,其中蕴含的信息将有助于人们挖掘和分析COVID-19病毒的传播模式。在世界疫情仍在蔓延的情况下,对新冠病毒传染模式的挖掘技术将不仅有助于世界疫情传染情况的分析,也将对未来可能存在的其他类型的病毒传染模式进行模拟。众多周知,对于大多数具传染性的病毒,如果人们相距很近,就很有可能被感染,病毒在人与人之间的传播与其行动轨迹之间也有较强的联性。传染模式是我们对病毒传播的一系列定义,需要根据我们的定义从移动数据中查找受感染对象。研究病毒传染行为可以帮助人们了解病毒的传播速度,这具有重要的现实意义。
本试验设计重点研究基于轨迹数据的传染模式挖掘技术,根据移动对象的轨迹数据,依据相关传染模型,通过初始传染源以及相关参数的确定,确定病毒在人与人之间的传播情况。

2、技术要求

1、熟悉轨迹数据的特点以及基本处理算法;
2、了解病毒传播行为的特点;
3、使用基础语言进行相应的程序设计;

二、需求分析

根据已知的轨迹数据(轨迹点),生成对应的轨迹序列,确定传染源后,按照一定的额传染规则进行传染情况的结果挖掘。

三、源程序代码

// 代码很烂,轻喷。注:仅供参考学习使用。
#  coding ='utf-8'
import os
from os import listdir
import copy

"""
函数:readData函数

说明:数据读取函数;
        实现功能:把轨迹数据按照文件内容进行读入,并进行格式化排序
        将truck里的数据导入,并对其数据格式化排序(增序)
            格式化后的数据实例:[1, 1, 23.845089, 38.01847]...

Parameters:
    无
Returns:
    data -- type = ’list‘
    将处理好的轨迹数据进行返回
"""


def readData():
    # 读取truck文件中文件名,truck_file_list用于保存
    truck_file_list = listdir('truck')
    # 列表内容显示
    # print(truck_file_list)
    # 统计truck文件下txt文件数量
    total_num_truckfile = len(truck_file_list)
    # 初始化构建data二维空列表
    data = [[] for _ in range(total_num_truckfile)]

    # 循环打开truck文件夹里的文件,将每个txt文件中的数据导入data中
    for i in range(total_num_truckfile):
        file_name = os.getcwd() + '/truck/' + truck_file_list[i]
        # 测试输出filename路径
        # print(file_name)
        with open(file_name, mode='r', encoding='utf-8') as f:
            # 直接读入(数据量不大使可以使用)
            data_temp = f.readlines()
            for j in range(len(data_temp)):
                # 数据导入格式化(去除导入数据时每一行末尾的换行符和数据之间的',')
                data[i].append(data_temp[j].rstrip("\n").split(','))
    # 数据导入后是字符串形式,通过下面的循环,将数据类型转换为数据本身的类型(int/floatfor i in range(total_num_truckfile):
        for j in range(len(data[i])):
            data[i][j] = list(map(eval, data[i][j]))

    # 将data数据按序号(第一个值)排序
    data.sort(key=lambda x: x[0])
    # 测试数据内容输出
    # print(data)
    # 将导入并格式化好的数据以多维列表的形式返回
    return data


"""
函数:injectCheck函数

说明:传染模式挖掘函数;
        实现功能:当用户选择初始传染源编号后,将初始传染源加入inject_total_data(含有初始传染源信息)列表中,然后进行初始传染的挖掘
                将挖掘信息存入inject_data(不含初始传染源信息)列表中,然后对inject_data中的被感染对象继续进行传染挖掘
                该函数内部,模拟实现了病毒传染的潜伏期delay_time、接触时长range_time和感染距离inject_distance。
                1.潜伏期:采用占用的时间节点来表示,程序中用2个时间节点来表示潜伏期为2,
                  程序中判断2个时间节点后,被感染体是否还有轨迹数据,若有则保存,若无则跳过
                2.接触时长:采用path_data[i][j-2],path_data[i][j-1]和path_data[i][j]三个时间节点进行测算欧氏距离,
                  选择三个结果中最大值,将其与inject_distance比较,最大值小于感染距离,则发生感染行为
                3.inject_distance模拟感染距离,建议浮动值在0.0001-0.0002
Parameters:
    data -- type = ’list‘ :是轨迹数据集
    initnum -- type =int‘:用户输入的初始传染源编号
    
Returns:
    retData -- type = ’list‘
    将处理好的感染数据进行返回
"""


def injectCheck(data, initnum):
    path_data = data
    # 初始传染源编号
    first_num = initnum - 1
    # 占用时间节点数量来模拟潜伏期
    delay_time = 2
    # 定为接触时间(近距离接触的时间范围)
    range_time = 3
    # 模拟感染距离
    inject_distance = 0.00012
    # 初始化已携带病毒传染源列表(含初始传染源),并置空
    inject_total_data = []
    # 初始化被感染的传染源列表(不含初始传染源),并置空
    inject_data = []
    inject_total_data.append(path_data[first_num][0])
    # print(inject_total_data)
    temp_record = []
    record = []

    # 进行初始传染源的传染挖掘
    count = 0
    for i in range(len(path_data)):
        for j in range(2, min(len(path_data[i]) - 1, len(path_data[first_num]))):
            """
                增加三个时间节点内的距离判断(前后各一个),调整循化范围,使得从第二个时间点开始判断到倒数第二个时间点,
                然后算三个时间点内的距离,如果最大的时间节点的位置都可以满足要求,其前后各个时间点位置的也都可以满足时距离要求
                此时保存最先的那个点即可
            """
            temp_distance_x1 = (path_data[i][j - 2][2] - path_data[first_num][j - 2][2]) ** 2
            temp_distance_y1 = (path_data[i][j - 2][3] - path_data[first_num][j - 2][3]) ** 2
            temp_distance1 = (temp_distance_x1 + temp_distance_y1) ** 0.5
            temp_distance_x2 = (path_data[i][j - 1][2] - path_data[first_num][j - 1][2]) ** 2
            temp_distance_y2 = (path_data[i][j - 1][3] - path_data[first_num][j - 1][3]) ** 2
            temp_distance2 = (temp_distance_x2 + temp_distance_y2) ** 0.5
            temp_distance_x3 = (path_data[i][j][2] - path_data[first_num][j][2]) ** 2
            temp_distance_y3 = (path_data[i][j][3] - path_data[first_num][j][3]) ** 2
            temp_distance3 = (temp_distance_x3 + temp_distance_y3) ** 0.5
            # 测算距离,选择最大值
            temp_distance = max(temp_distance1, temp_distance2, temp_distance3)
            # 将最大值与感染距离进行比较,比其小则发生传染行为。temp_distance != 0.0,是为了去除传染源本身
            if (temp_distance <= inject_distance) & (temp_distance != 0.0):
                temp_record.append([path_data[i][j][0], path_data[first_num][j][0], path_data[i][j][1], temp_distance])
                ret_dic = list(set([tuple(t) for t in temp_record]))
                record = [list(v) for v in ret_dic]
                record.sort(key=lambda x: [x[0], x[1]])
                # print(temp_record)
                count = count + 1
                # 将被初始传染源感染的数据添加到inject_total_data中
                inject_total_data.append(path_data[i][j])
                # 此处加判断潜伏期内容,判读时间点加delay_time后是否还在即可,如果不在,就直接去掉了
                if (path_data[i][j][1] + delay_time) < (len(path_data[i]) - 1):
                    inject_data.append(path_data[i][j + delay_time])
                else:
                    break
                break

    while (True):
        # 长度标记点,当去重后的inject_data长度不变(为0)时退出while循环
        flag_len = 0
        # 循环更新inject_data长度记录
        temp_len = len(inject_data)

        # 在已有的inject_data数据集中进行模式挖掘
        for i in range(len(inject_data)):
            # 传染源数据的编号
            flag_num = inject_data[i][0] - 1
            # 传染源被感染时间
            flag_inject_time = inject_data[i][1]
            for i2 in range(len(path_data)):
                if (len(path_data[i2]) > flag_inject_time):
                    for j2 in range(flag_inject_time,
                                    min(len(path_data[flag_num]), len(path_data[i2]) - 1) - 2):
                        temp_distance_x1 = (path_data[i2][j2 - 2][2] - path_data[flag_num][j2 - 2][2]) ** 2
                        temp_distance_y1 = (path_data[i2][j2 - 2][3] - path_data[flag_num][j2 - 2][3]) ** 2
                        temp_distance1 = (temp_distance_x1 + temp_distance_y1) ** 0.5
                        temp_distance_x2 = (path_data[i2][j2 - 1][2] - path_data[flag_num][j2 - 1][2]) ** 2
                        temp_distance_y2 = (path_data[i2][j2 - 1][3] - path_data[flag_num][j2 - 1][3]) ** 2
                        temp_distance2 = (temp_distance_x2 + temp_distance_y2) ** 0.5
                        temp_distance_x3 = (path_data[i2][j2][2] - path_data[flag_num][j2][2]) ** 2
                        temp_distance_y3 = (path_data[i2][j2][3] - path_data[flag_num][j2][3]) ** 2
                        temp_distance3 = (temp_distance_x3 + temp_distance_y3) ** 0.5
                        temp_distance = max(temp_distance1, temp_distance2, temp_distance3)
                        # print(path_data[i2][j2], '与', path_data[flag_num][j2], '的距离为:', temp_distance)
                        if (temp_distance <= inject_distance) & (temp_distance != 0.0):
                            temp_record.append(
                                [path_data[i2][j2][0], path_data[flag_num][j2][0], path_data[i2][j2][1], temp_distance])
                            ret_dic = list(set([tuple(t) for t in temp_record]))
                            record = [list(v) for v in ret_dic]
                            record.sort(key=lambda x: [x[0], x[1]])
                            # print(record)
                            # 将被传染源感染的数据添加到inject_total_data中
                            inject_total_data.append(path_data[i2][j2])
                            # 去重
                            dic = list(set([tuple(t) for t in inject_total_data]))
                            inject_total_data = [list(v) for v in dic]
                            # 排序
                            inject_total_data.sort(key=lambda x: [x[0], x[1]])
                            # 此处加判断潜伏期内容,判读时间点加delay_time后是否还在即可,如果不在,就直接去掉了
                            if (path_data[i2][j2][1] + delay_time) < (len(path_data[i2]) - 1):
                                inject_data.append(path_data[i2][j2 + delay_time])
                                # 去重
                                dic = list(set([tuple(t) for t in inject_data]))
                                inject_data = [list(v) for v in dic]
                                # 排序
                                inject_data.sort(key=lambda x: [x[0], x[1]])
                            else:
                                break
                            flag_len = temp_len - len(inject_data)

                else:
                    break
        if flag_len == 0:
            break

    recordData = setrecord(record)
    retData = setData(inject_total_data)
    return retData, recordData


"""
函数:setData函数

说明:数据处理函数;
        实现功能:把挖掘后简单去重的数据结果进行深度去重和排序
        例如,当injectdata=[[1,2,x.xxx,x.xxx],[1,4,x.xxx,x.xxx],[2,3,x.xxx,x.xxx]...]这样的数据时
        通过本函数实现保留:[[1,2,x.xxx,x.xxx],[2,3,x.xxx,x.xxx]...]的结果
        
Parameters:
    injectdata -- type = ’list‘
    接收一个数据列表,进行函数内部操作

Returns:
    li -- type = ’list‘
    将处理好的感染数据进行返回
"""


def setData(injectdata):
    li = []
    li.append(injectdata[0])
    for i in range(1, len(injectdata)):
        if injectdata[i][0] != injectdata[i - 1][0]:
            li.append(injectdata[i])
        else:
            continue

    return li


def setrecord(record):
    li = []
    if len(record) != 0:
        li.append(record[0])
        for i in range(1, len(record)):
            if (record[i][0] != record[i - 1][0]):
                li.append(record[i])
            else:
                continue
    return li


"""
函数:inject函数
说明:程序运行的主函数,执行步骤:
        1.先调用readData()载入truck数据文件下的轨迹数据信息
        2.由用户输入想要作为初始传染源的编号
        3.将(1)中载入的数据和初始传染源编号作为参数,调用injeCheck()函数进行传染模式挖掘
        4.将挖掘返回的信息格式化显示

Parameters:
    None
Returns:
    None
"""


def inject():
    path_data = readData()

    # 用户输入
    init_num = int(input('请输入初始传染源标号(1-276):').strip())
    print("正在处理中...")

    # 进行传染挖掘injectData接收挖掘后返回的数据
    injectData, record_data = injectCheck(path_data, init_num)

    for i in range(len(record_data)):
        print(record_data[i][0], '号,在时间点', record_data[i][2], '处,被', record_data[i][1], '号传染,传染距离为:', record_data[i][3])
    # injectData = setData(injectData)

    print('共计新感染{}人!'.format(len(injectData) - 1))
    # for num in injectData:
    print('所有携带病毒的编号为:', *[num[0] for num in injectData])

    # get_data(path_data,injectData)
    total_time_node = []
    for i in range(len(injectData)):
        print(injectData[i])
        total_time_node.append(path_data[injectData[i][0] - 1][injectData[i][1] - 1:])
    for i in range(len(total_time_node)):
        print(" injectDtat中第", i + 1, "个数据:", injectData[i], "其后续完整数据为:\n", total_time_node[i])


"""
函数:’main‘
说明:程序运行入口,调用inject函数进行运行

Parameters:
    None 
Returns:
    None  
"""
if __name__ == '__main__':
    inject()

四、测试总结

程序运行的部分就不贴图啦。有兴趣的话,大家自己运行一下吧~

五、其他

以上以及博客中含有在校整理字样的博客内容,均是本人本科在校学习期间所写的代码,源文件以及数据文件等在个人主页资源区可以获取。本人水平很低,大佬轻喷。整理出来是为了帮助初学者学习的过程中,在有需要的情况下能够有所参考。再次声明:仅供学习交流使用。大家有什么问题,欢迎留言~

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值