在python中使用csv文件数据绘制热图

cngwdvgl  于 2021-09-08  发布在  Java
关注(0)|答案(0)|浏览(250)

我已经输出了名为all\u count\u details\u dictionary的嵌套字典变量。使用该变量,我使用以下命令将数据保存到csv文件中。

import pandas as pd
csv_path = '../results_v6/output_01.csv'

# creating pandas dataframe using concat mehtod to extract data from dictionary

df = pd.concat([pd.DataFrame(l) for l in all_count_details_dictionary],axis=1).T

# saving the dataframe to the csv file

df.to_csv(csv_path, index=True)

输出的csv文件如下所示

因此,我使用以下代码绘制了一个图表

import matplotlib.pyplot as plt

def extract_csv_gen_plot(csv_path):

    length = 1503 #len(dataframe_colums_list)
    data = np.genfromtxt(csv_path, delimiter=",", skip_header=True, usecols=range(3, (length+1)))
    print(data)

    # renaming data axes
    #fig, ax = plt.subplots()
    #fig.canvas.draw()

    #labels =[item.get_text() for item in ax.get_xticklabels()]
    #labels[1] = 'testing'
    #ax.set_xticklabels(labels)

    #ax.set_xticklabels(list)
    #ax.set_yticklabels(list)
    #plt.setp(ax.get_xticklabels(), rotation = 90)

    plt.imshow(data, cmap='hot',interpolation='nearest')
    plt.show()

我试图将列标签和案例详细信息标签放入图形轴中,但没有成功。有没有人能告诉我,除了这个,还有什么其他最好的方法可以把这个表绘制成热图?
非常感谢。

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题