Hi! This is my code for the Orion constellation project from the Data Visualization Path.
%matplotlib notebook
from matplotlib import pyplot as plt
from matplotlib import colors as mcolors
from mpl_toolkits.mplot3d import Axes3D
Orion
x = [-0.41, 0.57, 0.07, 0.00, -0.29, -0.32,-0.50,-0.23, -0.23]
y = [4.12, 7.71, 2.36, 9.10, 13.35, 8.13, 7.19, 13.25,13.43]
z = [2.06, 0.84, 1.56, 2.07, 2.36, 1.72, 0.66, 1.25,1.38]
#2D plot
fig = plt.figure(figsize=(10,8))
ax = fig.add_subplot(1,1,1)
plt.scatter(x,y, color = âwhiteâ, marker="*")
plt.title(âOrion constellationâ)
plt.xlabel(âCoordinate xâ)
plt.ylabel(âCoordinate yâ)
ax.set_facecolor(âxkcd:dark blueâ)
plt.show()
#3D plot
fig_3D = plt.figure(figsize = (10,8))
constellation3d = fig_3D.add_subplot(1,1,1, projection = â3dâ)
constellation3d.scatter(x,y,z, color = âwhiteâ, marker="*", s=100)
plt.title(âOrion constellationâ)
plt.xlabel(âCoordinate xâ)
plt.ylabel(âCoordinate yâ)
constellation3d.set_zlabel(âCoordinate zâ)
plt.gca().patch.set_facecolor(âwhiteâ)
constellation3d.xaxis.set_pane_color((0.0, 0.011764705882352941, 0.3568627450980392, 1.0))
constellation3d.yaxis.set_pane_color((0.0, 0.011764705882352941, 0.3568627450980392, 1.0))
constellation3d.zaxis.set_pane_color((0.0, 0.011764705882352941, 0.3568627450980392, 1.0))
constellation3d.grid(False)
plt.show()
plt.savefig(âconstellation_3D.pngâ)