多进程使用matplotlib.pyplot绘heatmap（多线程不可以）

时间：2021-08-26 07:52:51

数据格式如下：

8_15/l_eye/2732.png -20.5773 -5.17769 -3.34583 21.5859

9_13_1/l_eye/1211.png -10.1145 34.9928 -38.2122 -26.3371

8_20/l_eye/5966.png -44.0264 50.2898 63.5838 -49.1353

8_13/l_eye/8780.png -16.9358 50.4528 -44.2617 -57.1462

9_16_2/l_eye/5370.png -21.2264 17.0589 4.33619 -20.3562

9_15_1/l_eye/66.png 40.5758 -21.0923 12.0032 40.8452

8_13/l_eye/6664.png 51.0789 55.3987 -67.2433 -79.1243

9_15_2/l_eye/4429.png 16.958 30.0386 -24.5935 -26.4802

8_21/l_eye/2579.png -20.619 4.7845 21.9891 27.529

8_21/l_eye/8464.png -36.8559 54.4664 -32.1576 -67.6335

8_21/l_eye/359.png 20.9732 2.25414 -3.88966 41.175

9_16_2/l_eye/3065.png 7.16623 43.091 35.9651 -28.4994

9_14_2/l_eye/1961.png 33.3302 28.3553 22.7904 -28.5209

9_16_1/l_eye/2038.png 56.9721 24.6518 -23.5831 -39.2209

以2、3列为x、y绘制一个热力图

以4、5列为x、y绘制一个热力图

#!/usr/bin/python

# -*- encoding: utf-8 -*-

import numpy as np

from matplotlib import pyplot as plt

#import thread

#from threading import Thread

from multiprocessing import Process

import pdb

def generate_heat_array(is_test=0):

    #pdb.set_trace()

    if is_test==1:

        # gussian distribution

        mean = [0,0]

        cov = [[0,1],[1,0]]

        x, y = np.random.multivariate_normal(mean, cov, 10000).T

        show_heat_map(x,y)

        return

    x_head=[]

    y_head=[]

    x_gaze=[]

    y_gaze=[]

    for line in open('train.txt'):

        split_data=line.split()

        x_head.append(float(split_data[1]))

        y_head.append(float(split_data[2]))

        x_gaze.append(float(split_data[3]))

        y_gaze.append(float(split_data[4]))

    #用thread库实现多线程

    #由于主线程退出时，子线程自动中止，因此需要join；由于thread库未提供join方法，所以需要自己手动实现。

    #thread.start_new_thread(show_heat_map,(x_head,y_head,1))

    #thread.start_new_thread(show_heat_map,(x_gaze,y_gaze,2))

    #用threading库实现多线程

    #threading库提供了join方法。但是由于matplotlib.pyplot中的方法都是全局的，因此用多线程绘图会有错误：RuntimeError: main thread is not in main loop

    #head_thread=Thread(target=show_heat_map, args=(x_head,y_head,1,))

    #gaze_thread=Thread(target=show_heat_map,args=(x_gaze,y_gaze,2,))

    #head_thread.start()

    #gaze_thread.start()

    #head_thread.join()

    #gaze_thread.join()

    #用multiprocessing实现多进程

    head_process=Process(target=show_heat_map,args=(x_head,y_head,1,))

    gaze_process=Process(target=show_heat_map,args=(x_gaze,y_gaze,2,))

    head_process.start()

    gaze_process.start()

    head_process.join()

    gaze_process.join()

def show_heat_map(x,y,n):

    #pdb.set_trace()

    fig=plt.figure(n)

    plt.hist2d(x,y,bins=100)

    plt.grid(True)

    plt.colorbar()

    #fig.savefig('%02i.png'%n)

    plt.show()

if __name__=='__main__':

    generate_heat_array(0)

绘制热力图的方法：

    plt.hist2d(x,y,bins=100)

x为横轴的值的list，y为纵轴值的list

修改bins可以控制区间大小

参考：http://blog.topspeedsnail.com/archives/707

使用meshgrid+imshow的话横纵坐标会有问题

相关文章