FYS3150/Project-3
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Made some changes

Browse Source
This commit is contained in:
Cory Balaton 2023-10-22 15:19:45 +02:00
parent 0dfda79fda
commit e42c108d20
No known key found for this signature in database
GPG Key ID: 3E5FCEBFD80F432B
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/scripts/plot_2_particles.py
View File

@ -34,12 +34,17 @@ def main():
"seagreen", "seagreen",
"darkred" "darkred"
] ]
start_pos = set()
for label, color, file in zip(labels, colors, files): for label, color, file in zip(labels, colors, files):
with open(file) as f: with open(file) as f:
lines = f.readlines() lines = f.readlines()
t = np.linspace(0, 50, len(lines)) t = np.linspace(0, 50, len(lines))
r = np.array([list(map(float, line.strip().split(","))) for line in lines]) r = np.array([list(map(float, line.strip().split(","))) for line in lines])
plt.plot(r[:,0], r[:,1], label=label, color=color) plt.plot(r[:,0], r[:,1], label=label, color=color)
start_pos.add((r[0,0],r[0,1]))
for pos in start_pos:
plt.plot(*pos, "o", color="black")
plt.xlabel(r"x $(\mu m)$") plt.xlabel(r"x $(\mu m)$")
plt.ylabel(r"y $(\mu m)$") plt.ylabel(r"y $(\mu m)$")