如何将csv文件的列转换为矩阵并为其绘制图?

时间:2019-06-11 09:58:21

标签: python matrix heatmap holoviews

我有一个csv文件。我只想将其中的一列分开,并开发一个矩阵,并使用该矩阵绘制热图图。

例如,如果我在列中有20行,我想制作10x2矩阵,这意味着矩阵中每行10列(在csv文件中从第一行到第十行),然后接下来的10行应显示为第二行的列。

然后,我想使用代表矩阵的holoviews绘制热图图。

1 个答案:

答案 0 :(得分:0)

首先,将您的csv文件读入数据框,如下所示。

import pandas as pd
df = pd.read_csv('filename.csv')

出于示例目的,这是一个包含伪数据的数据框

df = pd.DataFrame({'a':list(range(0,20)),'b':list(range(20,40))})

现在抓取列“ b”,并将其重塑为10x2。

mat = df['b'].values.reshape(10,2)

最后,使用一些虚拟名称生成热图。

import holoviews as hv
hv.extension('bokeh')
hv.HeatMap((['row','names'],list(range(0,10)),mat))