Python中为朋友画图操作

╰+哭是因爲堅強的太久メ 2022-12-25 06:53 176阅读 0赞

import numpy as np
    import pandas as pd
    import matplotlib.pyplot as plt
    import matplotlib as mpl
    
    if __name__ == '__main__':
        mpl.rcParams['font.family'] = 'STSong'
        mpl.rcParams['axes.unicode_minus'] = False
        # 选择的分别是-注视时间，-首次注视时间，-注视次数
        selected = ['IA_DWELL_TIME', 'IA_FIRST_FIXATION_DURATION', 'IA_FIXATION_COUNT']
        selected_chi = ['平均注视时间','首次注视时间','注视次数']
        for count in range(1,2):
            selected_img = 'Y' + str(count) + ".JPG"
            all_data = pd.read_excel('./zlj5.xlsx')
            # print(all_data)
            all_mean = []
            all_var = []
            all_SelectedImg = all_data[all_data['image'] == selected_img]
            all1Img = all_SelectedImg[all_SelectedImg['IA_LABEL'] == 1]
            all2Img = all_SelectedImg[all_SelectedImg['IA_LABEL'] == 2]
            all3Img = all_SelectedImg[all_SelectedImg['IA_LABEL'] == 3]
            all4Img = all_SelectedImg[all_SelectedImg['IA_LABEL'] == 4]
            # , all3Img, all4Img
            data = [all1Img,all2Img,all3Img,all4Img]
            for i in range(len(data)):
                for index, name in enumerate(selected):
                    if name != "IA_FIRST_FIXATION_DURATION":
                        tmp = np.array(data[i][name].tolist())
                    else:
                        tmp = data[i][data[i][name] != '.']
                        tmp = np.array(tmp[name].tolist())
                    all_mean.append(tmp.mean() / tmp.max())
            plotData = np.array([all_mean[i : i + 3] for i in range(0, len(all_mean),3)])
            # print(plotData)
            # print(plotData[:,0])
            # print(mpl.rcParamsDefault)
    
    
            width = 0.2 # the width of bar
            ind = np.arange(plotData.shape[0]) # the x locations for the groups
            fig, ax = plt.subplots()
            ax.bar(ind - 3 * width / 2, plotData[:,0], width, color='#D3D3D3', label='平均注视时间')
            ax.bar(ind - 1 * width / 2, plotData[:,1], width, color='#FFB6C1', label='首次注视时间')
            ax.bar(ind + 1 * width / 2, plotData[:,2], width, color='#12B4E5', label='注视次数')
            ax.set_ylabel('平均值与最大值比值(%)')
            ax.set_title(selected_img,pad=20)
            print(ind-1 * width / 2)
            print(ind)
            plt.xticks(ind - 1 * width / 2, ('分组1', '分组2','分组3','分组4'))
            plt.legend(loc=2, bbox_to_anchor=(1,1),borderaxespad = 0.)
            plt.subplots_adjust(right=0.78)
            plt.savefig("C:/Users/珍惜/Desktop/PythonMachineLearning/20201109/zlj3/" +'Y' + str(count) + '/'+selected_img+'.jpg')
            plt.show()