3dplot

3dplot.py

#! /usr/bin/env python

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

file = open("map1.txt")
n = int(file.readline()) + 1
hei = []
for line in file:
    hei.append([float(x) for x in line.strip().split()])
Z = np.array(hei).reshape((n, n))
x = range(n)
X, Y = np.meshgrid(x, x)
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm)
plt.colorbar(surf, shrink=0.7)

plt.show()

im = plt.imshow(Z, cmap=cm.coolwarm)
plt.colorbar(im, shrink=0.7)

plt.show()