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

Coryab/code #10

Merged
coryab merged 37 commits from coryab/code into develop 2023-10-24 10:45:33 +00:00
Conversation 0 Commits 37 Files Changed 128 +5
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit e42c108d20 - Show all commits

5
src/scripts/plot_2_particles.py
View File

@ -34,13 +34,18 @@ 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)$")
plt.legend(loc="upper right") plt.legend(loc="upper right")