klotz: matplotlib* + heatmap*

0 bookmark(s) - Sort by: Date ↓ / Title / - Bookmarks from other users for this tag

  1. import numpy as np import matplotlib.pyplot as plt from scipy.interpolate import griddata

    Load data from CSV

    dat = np.genfromtxt('dat.xyz', delimiter=' ',skip_header=0) X_dat = dat :,0 » Y_dat = dat :,1 » Z_dat = dat :,2 »

    Convert from pandas dataframes to numpy arrays

    X, Y, Z, = np.array( » ), np.array( » ), np.array( » ) for i in range(len(X_dat)): X = np.append(X,X_dat i » ) Y = np.append(Y,Y_dat i » ) Z = np.append(Z,Z_dat i » )

    create x-y points to be used in heatmap

    xi = np.linspace(X.min(),X.max(),1000) yi = np.linspace(Y.min(),Y.max(),1000)

    Z is a matrix of x-y values

    zi = griddata((X, Y), Z, (xi None,: » , yi :,None » ), method='cubic')

    I control the range of my colorbar by removing data

    outside of my range of interest

    zmin = 3 zmax = 12 zi (zi<zmin) | (zi>zmax) » = None

    Create the contour plot

    CS = plt.contourf(xi, yi, zi, 15, cmap=plt.cm.rainbow, vmax=zmax, vmin=zmin) plt.colorbar()
    plt.show()

    2017-07-30 Tags: , , , , by klotz

Top of the page

First / Previous / Next / Last / Page 1 of 0 SemanticScuttle - klotz.me: Tags: matplotlib + heatmap

About - Propulsed by SemanticScuttle