Part 2 - Visualization This part is worth 3 points Our print atoms function is a

Question

Part 2 - Visualization This part is worth 3 points Our print atoms function is a very poor way to show our atoms on the screen Create a new function: def plot atoms(atoms, container radius, filename): This function will plot atoms on a matplotlib plot and writes the file atoms: a 2d numpy array with x.y coordinates of atoms filename: the name of the file to write out pass Your function should: Draw a circle onto the plot with the same radius as the container (with a colour to be selected by you) Draw atoms onto the plot (with a plot marker and colour to be selected by you) Here is a sample output (yours won't look exactly the same): Atoms in a beaker 10.0 7.5 5.0 2.5 0.0 -2.5 -5.0 -10.0 -10 0 10

Explanation / Answer

import numpy as np
import matplotlib.pyplot as plt

def plot_atoms(atoms,container_radius,filename):
circle= plt.Circle((0,0), radius= container_radius, color='green') #Circle with green colour
ax=plt.gca()
ax.add_patch(circle)
plt.axis('scaled')
plt.plot(atoms, 'ro') #atoms with red colour
plt.savefig(filename)
plt.show()

#Sample example, it plots atoms in red colour and circle in green colour with radius 5 and saves the image in #image1.jpeg
plot_atoms(np.array([[1,2],[3,3],[4,4],[5,2]]),5,"image1.jpeg")

Related Questions

Navigate

• Browse (All)
• Browse P
• Subjects

---

---

Integrity-first tutoring: explanations and feedback only — we do not complete graded work. Learn more.