Codecademy Forums

Data Science: Constellation

Hi everyone! Here is my take on the Orion contellation project.

Feedback welcome!

CODE

enable rotation

%matplotlib notebook

libraries

from matplotlib import pyplot as plt
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]

create 2D figure

orion = plt.figure()

save it to a variable to change colors later

ax = orion.add_subplot(1,1,1)

plot with pretty star markers

plt.scatter(x,y,color=“white”,marker=’*’)

change background colors

ax = plt.gca()
ax.set_facecolor(‘black’)

title

plt.title(‘Orion in 2D’)

display graph

plt.show()

save graph

plt.savefig(‘orion_2D.png’)

orion_2D

create figure

fig_3d = plt.figure()

save it to a variable to change colors later

ax3 = fig_3d.add_subplot(1,1,1,projection=‘3d’)

plot with pretty star markers

constellation3d = plt.scatter(x,y,z,color=“red”,marker=’*’)

change background colors

ax3 = plt.gca()
ax3.set_facecolor(‘black’)
ax3.tick_params(axis=‘x’, colors=‘lightgray’)
ax3.tick_params(axis=‘y’, colors=‘lightgray’)
ax3.tick_params(axis=‘z’, colors=‘lightgray’)

#title
plt.title(‘Orion in 3D’,color=‘white’)

#display graph
plt.show()

#save graph
plt.savefig(‘orion_3D.png’)

1 Like