python 输入列的名字，直接可以得到图像

最新推荐文章于 2022-08-17 22:57:05 发布

weixin_30858241

最新推荐文章于 2022-08-17 22:57:05 发布

阅读量107

点赞数

文章标签： python

原文链接：http://www.cnblogs.com/dotiger/p/8894196.html

版权

# data_dict = {  #Time  q32    q34    q42    q44    rtouch1    rtouch2    rtouch3    rtouch4    ltouch1    ltouch2    ltouch3
# ltouch4  vec_hip_lx vec_knee_lx    vec_hip_rx vec_knee_rx    acc_hip_lx_angle   acc_knee_lx_angle
#  acc_hip_rx_angle   acc_knee_rx_angle
#
#     'Time':[],
#     'q32':[],

# }
import time
import matplotlib.pyplot as plt

fig= plt.subplot()


def  data_dictionary (path,data_dict = {}):
    data = open(path,'r',encoding= 'utf-8')
    for line in data :
        #print(line)
        line_list = line.strip().split('\t')
        print(line_list)
        for item in line_list:
            data_dict[item] = []
        colum_name = line_list
        #print('111111111111111111111111111', data_dict)

        break

    for line in data :
         line_list = line.strip().split('\t')
         #print(line_list)
         for i in  range(len(line_list)):
            # print(colum_name[i])
             #print(line_list[i])
            data_dict[colum_name[i]].append(float(line_list[i]))
         #print(data_dict)

            #data_dict[str(colum_name[i])].append(line_list[i])
            #print(i)

    return data_dict
def  drawing_pitcure (path,x_axis,*args):
    data_dict= data_dictionary(path)
    for item in  args:
        if  item in  data_dict.keys():

            fig.plot(data_dict[x_axis],data_dict[item])
            print('x_axis',data_dict[x_axis])
            print('data_dict',data_dict[item])


        else :
            print('there is no this variable')
    plt.show()





data_dict = data_dictionary(r'C:\Users\LENOVO\Desktop\python\juemi\0411\OutputAngles3.xls')

drawing_pitcure(r'C:\Users\LENOVO\Desktop\python\juemi\0411\OutputAngles3.xls','Time','ltouch2','rtouch3')