使用Matplotlib的热图以及三列csv中的数据

时间:2019-03-10 07:29:50

标签: python matplotlib heatmap

使用Python matplotlib,我需要从一个csv文件构造一个热图,其中第一列和第二列分别具有x和y变量。第三列具有z变量(它是x和y的函数)。我想使用2D热图在颜色上表示z变量。

csv文件中的x和y变量没有特定顺序。

我开始研究一些代码以从文件中创建矩阵,但是没有继续进行。

def heat_map_simple(csv_file,x_var,y_var, color_var):

    data = pd.read_csv(csv_file)
    data = pd.DataFrame(data)
    x_n = len(set(data.iloc[:][x_var]))
    y_n = len(set(data.iloc[:][y_var]))
    matrix = np.zeros((x_n, y_n))
    rows = data.shape[1]
    for r in xrange(rows):
        val = data.iloc[r][color_var]

0 个答案:

没有答案