Add forbidden z region#
import molsysmt as msm
import openmm as mm
from openmm import unit
from openmm import app
from tqdm import tqdm
import numpy as np
from matplotlib import pyplot as plt
system = mm.System()
system.addParticle(39.948 * unit.amu) # masa del átomo de argón
0
msm.thirds.openmm.forces.add_forbidden_z_region(system, z0='1.0 nm', width='1.5 nm',
force_constant = '5000 kilojoules/(mol*angstroms**2)',
pbc=True)
0
# Formalismo NVT
temperature = 300*unit.kelvin
integration_timestep = 2.0*unit.femtoseconds
saving_timestep = 1.00*unit.picoseconds
simulation_time = 1000.*unit.picoseconds
saving_steps = int(saving_timestep/integration_timestep)
num_saving_steps = int(simulation_time/saving_timestep)
friction = 5.0/unit.picoseconds
integrator = mm.LangevinIntegrator(temperature, friction, integration_timestep)
platform = mm.Platform.getPlatformByName('OpenCL')
times = np.zeros(num_saving_steps, np.float32) * unit.picoseconds
positions = np.zeros([num_saving_steps,3], np.float32) * unit.nanometers
velocities = np.zeros([num_saving_steps,3], np.float32) * unit.nanometers/unit.picosecond
potential_energies = np.zeros([num_saving_steps], np.float32) * unit.kilocalories_per_mole
kinetic_energies = np.zeros([num_saving_steps], np.float32) * unit.kilocalories_per_mole
initial_positions = [[0.0, 0.0, 2.0]] * unit.nanometers
context = mm.Context(system, integrator, platform)
context.setPositions(initial_positions)
L = 2.0
v1 = [L,0,0] * unit.nanometers
v2 = [0,L,0] * unit.nanometers
v3 = [0,0,L] * unit.nanometers
L = L * unit.nanometers
context.setPeriodicBoxVectors(v1, v2, v3)
state = context.getState(getEnergy=True, getPositions=True, getVelocities=True)
times[0] = state.getTime()
positions[0] = state.getPositions()[0]
velocities[0] = state.getVelocities()[0]
kinetic_energies[0]=state.getKineticEnergy()
potential_energies[0]=state.getPotentialEnergy()
for ii in tqdm(range(1,num_saving_steps)):
context.getIntegrator().step(saving_steps)
state_xx = context.getState(getEnergy=True, getPositions=True, getVelocities=True)
times[ii] = state_xx.getTime()
positions[ii] = state_xx.getPositions()[0]
velocities[ii] = state_xx.getVelocities()[0]
kinetic_energies[ii]=state_xx.getKineticEnergy()
potential_energies[ii]=state_xx.getPotentialEnergy()
0%| | 0/999 [00:00<?, ?it/s]
0%| | 2/999 [00:00<00:59, 16.74it/s]
1%| | 5/999 [00:00<00:48, 20.69it/s]
1%| | 8/999 [00:00<00:47, 21.03it/s]
1%| | 11/999 [00:00<00:48, 20.48it/s]
1%|▏ | 14/999 [00:00<00:48, 20.45it/s]
2%|▏ | 17/999 [00:00<00:50, 19.51it/s]
2%|▏ | 19/999 [00:00<00:50, 19.43it/s]
2%|▏ | 21/999 [00:01<00:50, 19.29it/s]
2%|▏ | 23/999 [00:01<00:51, 18.81it/s]
3%|▎ | 25/999 [00:01<00:51, 18.86it/s]
3%|▎ | 27/999 [00:01<00:52, 18.56it/s]
3%|▎ | 30/999 [00:01<00:49, 19.74it/s]
3%|▎ | 32/999 [00:01<00:49, 19.71it/s]
4%|▎ | 35/999 [00:01<00:52, 18.22it/s]
4%|▎ | 37/999 [00:01<00:56, 17.07it/s]
4%|▍ | 39/999 [00:02<00:57, 16.73it/s]
4%|▍ | 41/999 [00:02<01:01, 15.48it/s]
4%|▍ | 44/999 [00:02<00:54, 17.47it/s]
5%|▍ | 46/999 [00:02<00:54, 17.64it/s]
5%|▍ | 48/999 [00:02<00:52, 17.97it/s]
5%|▌ | 50/999 [00:02<00:53, 17.59it/s]
5%|▌ | 52/999 [00:02<00:53, 17.79it/s]
5%|▌ | 54/999 [00:02<00:52, 17.90it/s]
6%|▌ | 56/999 [00:03<00:52, 17.89it/s]
6%|▌ | 58/999 [00:03<00:56, 16.73it/s]
6%|▌ | 60/999 [00:03<00:53, 17.46it/s]
6%|▋ | 63/999 [00:03<00:48, 19.47it/s]
7%|▋ | 66/999 [00:03<00:48, 19.43it/s]
7%|▋ | 68/999 [00:03<00:49, 18.91it/s]
7%|▋ | 70/999 [00:03<00:50, 18.34it/s]
7%|▋ | 72/999 [00:03<00:49, 18.68it/s]
7%|▋ | 74/999 [00:04<00:49, 18.68it/s]
8%|▊ | 76/999 [00:04<00:48, 18.94it/s]
8%|▊ | 79/999 [00:04<00:48, 19.16it/s]
8%|▊ | 81/999 [00:04<00:48, 18.95it/s]
8%|▊ | 83/999 [00:04<00:49, 18.41it/s]
9%|▊ | 85/999 [00:04<00:50, 17.93it/s]
9%|▊ | 87/999 [00:04<00:50, 17.91it/s]
9%|▉ | 89/999 [00:04<00:49, 18.34it/s]
9%|▉ | 91/999 [00:04<00:49, 18.28it/s]
9%|▉ | 94/999 [00:05<00:46, 19.29it/s]
10%|▉ | 96/999 [00:05<00:48, 18.63it/s]
10%|▉ | 99/999 [00:05<00:44, 20.25it/s]
10%|█ | 102/999 [00:05<00:46, 19.28it/s]
11%|█ | 105/999 [00:05<00:42, 21.24it/s]
11%|█ | 108/999 [00:05<00:45, 19.54it/s]
11%|█ | 111/999 [00:05<00:47, 18.85it/s]
11%|█▏ | 113/999 [00:06<00:46, 18.94it/s]
12%|█▏ | 116/999 [00:06<00:42, 20.68it/s]
12%|█▏ | 119/999 [00:06<00:43, 20.16it/s]
12%|█▏ | 122/999 [00:06<00:43, 20.01it/s]
13%|█▎ | 125/999 [00:06<00:43, 19.93it/s]
13%|█▎ | 128/999 [00:06<00:45, 19.08it/s]
13%|█▎ | 131/999 [00:06<00:43, 19.83it/s]
13%|█▎ | 134/999 [00:07<00:46, 18.76it/s]
14%|█▎ | 137/999 [00:07<00:46, 18.71it/s]
14%|█▍ | 139/999 [00:07<00:50, 17.08it/s]
14%|█▍ | 141/999 [00:07<00:49, 17.26it/s]
14%|█▍ | 143/999 [00:07<00:49, 17.36it/s]
15%|█▍ | 145/999 [00:07<00:50, 16.90it/s]
15%|█▍ | 147/999 [00:07<00:48, 17.40it/s]
15%|█▍ | 149/999 [00:08<00:50, 16.90it/s]
15%|█▌ | 151/999 [00:08<00:50, 16.71it/s]
15%|█▌ | 153/999 [00:08<00:50, 16.86it/s]
16%|█▌ | 155/999 [00:08<00:50, 16.80it/s]
16%|█▌ | 158/999 [00:08<00:46, 18.15it/s]
16%|█▌ | 161/999 [00:08<00:44, 18.87it/s]
16%|█▋ | 163/999 [00:08<00:44, 18.73it/s]
17%|█▋ | 166/999 [00:08<00:45, 18.19it/s]
17%|█▋ | 168/999 [00:09<00:44, 18.52it/s]
17%|█▋ | 170/999 [00:09<00:47, 17.31it/s]
17%|█▋ | 172/999 [00:09<00:47, 17.27it/s]
18%|█▊ | 175/999 [00:09<00:44, 18.66it/s]
18%|█▊ | 177/999 [00:09<00:44, 18.60it/s]
18%|█▊ | 179/999 [00:09<00:43, 18.74it/s]
18%|█▊ | 182/999 [00:09<00:41, 19.87it/s]
18%|█▊ | 184/999 [00:09<00:42, 19.26it/s]
19%|█▊ | 186/999 [00:10<00:42, 18.96it/s]
19%|█▉ | 188/999 [00:10<00:42, 18.93it/s]
19%|█▉ | 191/999 [00:10<00:42, 18.95it/s]
19%|█▉ | 194/999 [00:10<00:41, 19.21it/s]
20%|█▉ | 197/999 [00:10<00:38, 21.05it/s]
20%|██ | 200/999 [00:10<00:38, 20.62it/s]
20%|██ | 203/999 [00:10<00:42, 18.54it/s]
21%|██ | 206/999 [00:11<00:41, 19.29it/s]
21%|██ | 208/999 [00:11<00:41, 18.94it/s]
21%|██ | 210/999 [00:11<00:42, 18.57it/s]
21%|██ | 212/999 [00:11<00:41, 18.91it/s]
22%|██▏ | 215/999 [00:11<00:38, 20.28it/s]
22%|██▏ | 218/999 [00:11<00:38, 20.52it/s]
22%|██▏ | 221/999 [00:11<00:38, 20.21it/s]
22%|██▏ | 224/999 [00:11<00:40, 19.16it/s]
23%|██▎ | 226/999 [00:12<00:42, 18.40it/s]
23%|██▎ | 229/999 [00:12<00:42, 18.09it/s]
23%|██▎ | 231/999 [00:12<00:42, 17.95it/s]
23%|██▎ | 233/999 [00:12<00:42, 17.99it/s]
24%|██▎ | 235/999 [00:12<00:42, 17.87it/s]
24%|██▍ | 238/999 [00:12<00:37, 20.03it/s]
24%|██▍ | 241/999 [00:12<00:35, 21.42it/s]
24%|██▍ | 244/999 [00:12<00:35, 21.00it/s]
25%|██▍ | 247/999 [00:13<00:36, 20.74it/s]
25%|██▌ | 250/999 [00:13<00:37, 19.76it/s]
25%|██▌ | 252/999 [00:13<00:38, 19.55it/s]
25%|██▌ | 254/999 [00:13<00:38, 19.39it/s]
26%|██▌ | 257/999 [00:13<00:35, 20.66it/s]
26%|██▌ | 260/999 [00:13<00:34, 21.48it/s]
26%|██▋ | 263/999 [00:13<00:35, 20.45it/s]
27%|██▋ | 266/999 [00:14<00:34, 21.36it/s]
27%|██▋ | 269/999 [00:14<00:33, 22.04it/s]
27%|██▋ | 272/999 [00:14<00:32, 22.50it/s]
28%|██▊ | 275/999 [00:14<00:31, 22.66it/s]
28%|██▊ | 278/999 [00:14<00:31, 22.92it/s]
28%|██▊ | 281/999 [00:14<00:31, 23.09it/s]
28%|██▊ | 284/999 [00:14<00:35, 20.34it/s]
29%|██▊ | 287/999 [00:15<00:36, 19.34it/s]
29%|██▉ | 289/999 [00:15<00:37, 18.84it/s]
29%|██▉ | 291/999 [00:15<00:42, 16.74it/s]
29%|██▉ | 294/999 [00:15<00:41, 17.10it/s]
30%|██▉ | 296/999 [00:15<00:41, 17.07it/s]
30%|██▉ | 299/999 [00:15<00:35, 19.70it/s]
30%|███ | 302/999 [00:15<00:32, 21.16it/s]
31%|███ | 305/999 [00:15<00:33, 21.01it/s]
31%|███ | 308/999 [00:16<00:36, 19.14it/s]
31%|███ | 310/999 [00:16<00:37, 18.47it/s]
31%|███ | 312/999 [00:16<00:37, 18.36it/s]
32%|███▏ | 315/999 [00:16<00:35, 19.37it/s]
32%|███▏ | 317/999 [00:16<00:36, 18.85it/s]
32%|███▏ | 320/999 [00:16<00:32, 21.04it/s]
32%|███▏ | 323/999 [00:16<00:34, 19.35it/s]
33%|███▎ | 325/999 [00:17<00:35, 18.89it/s]
33%|███▎ | 327/999 [00:17<00:36, 18.44it/s]
33%|███▎ | 329/999 [00:17<00:39, 17.15it/s]
33%|███▎ | 331/999 [00:17<00:38, 17.47it/s]
33%|███▎ | 333/999 [00:17<00:36, 18.11it/s]
34%|███▎ | 336/999 [00:17<00:33, 19.68it/s]
34%|███▍ | 339/999 [00:17<00:30, 21.52it/s]
34%|███▍ | 342/999 [00:17<00:31, 21.19it/s]
35%|███▍ | 345/999 [00:18<00:29, 22.50it/s]
35%|███▍ | 348/999 [00:18<00:30, 21.12it/s]
35%|███▌ | 351/999 [00:18<00:33, 19.58it/s]
35%|███▌ | 354/999 [00:18<00:32, 19.88it/s]
36%|███▌ | 357/999 [00:18<00:31, 20.52it/s]
36%|███▌ | 360/999 [00:18<00:32, 19.56it/s]
36%|███▌ | 362/999 [00:18<00:33, 18.94it/s]
37%|███▋ | 365/999 [00:19<00:30, 20.63it/s]
37%|███▋ | 368/999 [00:19<00:29, 21.37it/s]
37%|███▋ | 371/999 [00:19<00:31, 19.67it/s]
37%|███▋ | 374/999 [00:19<00:32, 18.98it/s]
38%|███▊ | 376/999 [00:19<00:32, 19.12it/s]
38%|███▊ | 379/999 [00:19<00:30, 20.43it/s]
38%|███▊ | 382/999 [00:19<00:28, 21.41it/s]
39%|███▊ | 385/999 [00:20<00:27, 21.98it/s]
39%|███▉ | 388/999 [00:20<00:27, 22.12it/s]
39%|███▉ | 391/999 [00:20<00:29, 20.54it/s]
39%|███▉ | 394/999 [00:20<00:30, 19.73it/s]
40%|███▉ | 397/999 [00:20<00:31, 19.36it/s]
40%|███▉ | 399/999 [00:20<00:31, 18.84it/s]
40%|████ | 401/999 [00:20<00:31, 18.83it/s]
40%|████ | 404/999 [00:21<00:29, 20.10it/s]
41%|████ | 407/999 [00:21<00:31, 18.62it/s]
41%|████ | 410/999 [00:21<00:31, 18.72it/s]
41%|████▏ | 413/999 [00:21<00:29, 20.16it/s]
42%|████▏ | 416/999 [00:21<00:29, 20.04it/s]
42%|████▏ | 419/999 [00:21<00:30, 19.28it/s]
42%|████▏ | 422/999 [00:21<00:30, 19.14it/s]
43%|████▎ | 425/999 [00:22<00:27, 20.76it/s]
43%|████▎ | 428/999 [00:22<00:27, 20.96it/s]
43%|████▎ | 431/999 [00:22<00:28, 19.77it/s]
43%|████▎ | 434/999 [00:22<00:28, 19.77it/s]
44%|████▎ | 437/999 [00:22<00:30, 18.68it/s]
44%|████▍ | 439/999 [00:22<00:29, 18.84it/s]
44%|████▍ | 441/999 [00:22<00:29, 18.66it/s]
44%|████▍ | 443/999 [00:23<00:31, 17.54it/s]
45%|████▍ | 445/999 [00:23<00:33, 16.61it/s]
45%|████▍ | 448/999 [00:23<00:31, 17.68it/s]
45%|████▌ | 450/999 [00:23<00:30, 18.14it/s]
45%|████▌ | 452/999 [00:23<00:29, 18.55it/s]
45%|████▌ | 454/999 [00:23<00:28, 18.83it/s]
46%|████▌ | 456/999 [00:23<00:28, 19.13it/s]
46%|████▌ | 459/999 [00:23<00:27, 19.65it/s]
46%|████▌ | 462/999 [00:24<00:27, 19.80it/s]
46%|████▋ | 464/999 [00:24<00:27, 19.63it/s]
47%|████▋ | 466/999 [00:24<00:27, 19.54it/s]
47%|████▋ | 468/999 [00:24<00:27, 19.60it/s]
47%|████▋ | 470/999 [00:24<00:27, 19.44it/s]
47%|████▋ | 473/999 [00:24<00:25, 20.49it/s]
48%|████▊ | 476/999 [00:24<00:27, 18.97it/s]
48%|████▊ | 478/999 [00:24<00:27, 18.65it/s]
48%|████▊ | 480/999 [00:25<00:28, 18.18it/s]
48%|████▊ | 482/999 [00:25<00:27, 18.53it/s]
48%|████▊ | 484/999 [00:25<00:28, 17.95it/s]
49%|████▊ | 487/999 [00:25<00:26, 19.60it/s]
49%|████▉ | 489/999 [00:25<00:27, 18.32it/s]
49%|████▉ | 492/999 [00:25<00:26, 19.34it/s]
50%|████▉ | 495/999 [00:25<00:25, 19.64it/s]
50%|████▉ | 497/999 [00:25<00:26, 19.04it/s]
50%|████▉ | 499/999 [00:26<00:26, 18.93it/s]
50%|█████ | 502/999 [00:26<00:24, 20.57it/s]
51%|█████ | 505/999 [00:26<00:23, 20.71it/s]
51%|█████ | 508/999 [00:26<00:27, 17.72it/s]
51%|█████ | 510/999 [00:26<00:28, 17.22it/s]
51%|█████▏ | 513/999 [00:26<00:24, 19.66it/s]
52%|█████▏ | 516/999 [00:26<00:25, 19.16it/s]
52%|█████▏ | 518/999 [00:27<00:25, 18.68it/s]
52%|█████▏ | 520/999 [00:27<00:26, 18.42it/s]
52%|█████▏ | 522/999 [00:27<00:26, 18.23it/s]
52%|█████▏ | 524/999 [00:27<00:27, 17.45it/s]
53%|█████▎ | 526/999 [00:27<00:26, 17.70it/s]
53%|█████▎ | 528/999 [00:27<00:26, 17.69it/s]
53%|█████▎ | 530/999 [00:27<00:26, 17.84it/s]
53%|█████▎ | 532/999 [00:27<00:25, 17.98it/s]
53%|█████▎ | 534/999 [00:27<00:25, 18.09it/s]
54%|█████▍ | 537/999 [00:28<00:22, 20.16it/s]
54%|█████▍ | 540/999 [00:28<00:22, 20.02it/s]
54%|█████▍ | 543/999 [00:28<00:21, 21.14it/s]
55%|█████▍ | 546/999 [00:28<00:21, 21.28it/s]
55%|█████▍ | 549/999 [00:28<00:22, 20.45it/s]
55%|█████▌ | 552/999 [00:28<00:21, 20.41it/s]
56%|█████▌ | 555/999 [00:28<00:21, 20.75it/s]
56%|█████▌ | 558/999 [00:29<00:22, 19.21it/s]
56%|█████▌ | 560/999 [00:29<00:22, 19.21it/s]
56%|█████▋ | 562/999 [00:29<00:23, 18.92it/s]
57%|█████▋ | 565/999 [00:29<00:21, 20.29it/s]
57%|█████▋ | 568/999 [00:29<00:20, 20.53it/s]
57%|█████▋ | 571/999 [00:29<00:21, 19.52it/s]
57%|█████▋ | 573/999 [00:29<00:22, 19.13it/s]
58%|█████▊ | 575/999 [00:29<00:23, 18.28it/s]
58%|█████▊ | 577/999 [00:30<00:23, 18.14it/s]
58%|█████▊ | 579/999 [00:30<00:22, 18.45it/s]
58%|█████▊ | 582/999 [00:30<00:21, 19.70it/s]
58%|█████▊ | 584/999 [00:30<00:21, 19.73it/s]
59%|█████▊ | 586/999 [00:30<00:21, 19.03it/s]
59%|█████▉ | 588/999 [00:30<00:21, 19.08it/s]
59%|█████▉ | 590/999 [00:30<00:21, 18.77it/s]
59%|█████▉ | 593/999 [00:30<00:21, 19.27it/s]
60%|█████▉ | 596/999 [00:31<00:19, 20.20it/s]
60%|█████▉ | 599/999 [00:31<00:21, 18.57it/s]
60%|██████ | 602/999 [00:31<00:20, 19.48it/s]
61%|██████ | 605/999 [00:31<00:19, 20.16it/s]
61%|██████ | 608/999 [00:31<00:19, 20.13it/s]
61%|██████ | 611/999 [00:31<00:18, 20.69it/s]
61%|██████▏ | 614/999 [00:31<00:19, 20.22it/s]
62%|██████▏ | 617/999 [00:32<00:19, 19.89it/s]
62%|██████▏ | 620/999 [00:32<00:19, 19.27it/s]
62%|██████▏ | 622/999 [00:32<00:20, 18.74it/s]
62%|██████▏ | 624/999 [00:32<00:20, 18.37it/s]
63%|██████▎ | 626/999 [00:32<00:19, 18.72it/s]
63%|██████▎ | 628/999 [00:32<00:20, 18.53it/s]
63%|██████▎ | 630/999 [00:32<00:20, 18.36it/s]
63%|██████▎ | 633/999 [00:32<00:18, 19.45it/s]
64%|██████▎ | 635/999 [00:33<00:19, 18.95it/s]
64%|██████▍ | 637/999 [00:33<00:19, 18.46it/s]
64%|██████▍ | 639/999 [00:33<00:22, 16.36it/s]
64%|██████▍ | 641/999 [00:33<00:20, 17.21it/s]
64%|██████▍ | 643/999 [00:33<00:21, 16.88it/s]
65%|██████▍ | 646/999 [00:33<00:19, 18.34it/s]
65%|██████▍ | 649/999 [00:33<00:18, 18.46it/s]
65%|██████▌ | 651/999 [00:33<00:19, 18.00it/s]
65%|██████▌ | 653/999 [00:34<00:19, 17.66it/s]
66%|██████▌ | 656/999 [00:34<00:17, 19.30it/s]
66%|██████▌ | 659/999 [00:34<00:17, 19.75it/s]
66%|██████▌ | 661/999 [00:34<00:18, 18.21it/s]
66%|██████▋ | 663/999 [00:34<00:18, 18.04it/s]
67%|██████▋ | 665/999 [00:34<00:19, 17.52it/s]
67%|██████▋ | 667/999 [00:34<00:18, 17.53it/s]
67%|██████▋ | 669/999 [00:34<00:18, 17.83it/s]
67%|██████▋ | 671/999 [00:35<00:17, 18.39it/s]
67%|██████▋ | 674/999 [00:35<00:17, 18.52it/s]
68%|██████▊ | 676/999 [00:35<00:17, 18.44it/s]
68%|██████▊ | 678/999 [00:35<00:17, 18.43it/s]
68%|██████▊ | 680/999 [00:35<00:17, 18.20it/s]
68%|██████▊ | 683/999 [00:35<00:16, 19.52it/s]
69%|██████▊ | 685/999 [00:35<00:16, 19.47it/s]
69%|██████▉ | 687/999 [00:35<00:16, 19.38it/s]
69%|██████▉ | 689/999 [00:36<00:17, 17.92it/s]
69%|██████▉ | 692/999 [00:36<00:16, 18.44it/s]
69%|██████▉ | 694/999 [00:36<00:16, 18.37it/s]
70%|██████▉ | 696/999 [00:36<00:16, 17.87it/s]
70%|██████▉ | 698/999 [00:36<00:16, 18.21it/s]
70%|███████ | 700/999 [00:36<00:16, 18.30it/s]
70%|███████ | 702/999 [00:36<00:16, 18.49it/s]
70%|███████ | 704/999 [00:36<00:15, 18.54it/s]
71%|███████ | 706/999 [00:36<00:15, 18.39it/s]
71%|███████ | 708/999 [00:37<00:15, 18.63it/s]
71%|███████ | 710/999 [00:37<00:17, 16.59it/s]
71%|███████▏ | 712/999 [00:37<00:16, 16.98it/s]
72%|███████▏ | 715/999 [00:37<00:15, 18.31it/s]
72%|███████▏ | 717/999 [00:37<00:15, 18.27it/s]
72%|███████▏ | 719/999 [00:37<00:15, 18.67it/s]
72%|███████▏ | 721/999 [00:37<00:14, 18.67it/s]
72%|███████▏ | 723/999 [00:37<00:15, 17.84it/s]
73%|███████▎ | 726/999 [00:38<00:14, 19.23it/s]
73%|███████▎ | 728/999 [00:38<00:14, 18.84it/s]
73%|███████▎ | 730/999 [00:38<00:14, 18.51it/s]
73%|███████▎ | 732/999 [00:38<00:14, 18.23it/s]
73%|███████▎ | 734/999 [00:38<00:14, 18.45it/s]
74%|███████▎ | 736/999 [00:38<00:14, 18.04it/s]
74%|███████▍ | 738/999 [00:38<00:14, 18.37it/s]
74%|███████▍ | 740/999 [00:38<00:14, 18.21it/s]
74%|███████▍ | 742/999 [00:38<00:14, 18.31it/s]
74%|███████▍ | 744/999 [00:39<00:14, 18.12it/s]
75%|███████▍ | 746/999 [00:39<00:13, 18.09it/s]
75%|███████▍ | 748/999 [00:39<00:13, 17.97it/s]
75%|███████▌ | 750/999 [00:39<00:13, 18.09it/s]
75%|███████▌ | 752/999 [00:39<00:13, 18.16it/s]
75%|███████▌ | 754/999 [00:39<00:13, 18.27it/s]
76%|███████▌ | 756/999 [00:39<00:13, 18.24it/s]
76%|███████▌ | 758/999 [00:39<00:13, 18.17it/s]
76%|███████▌ | 760/999 [00:39<00:13, 17.93it/s]
76%|███████▋ | 762/999 [00:40<00:13, 17.77it/s]
76%|███████▋ | 764/999 [00:40<00:13, 17.73it/s]
77%|███████▋ | 767/999 [00:40<00:11, 19.87it/s]
77%|███████▋ | 770/999 [00:40<00:11, 19.55it/s]
77%|███████▋ | 772/999 [00:40<00:11, 19.41it/s]
77%|███████▋ | 774/999 [00:40<00:11, 19.22it/s]
78%|███████▊ | 776/999 [00:40<00:11, 19.09it/s]
78%|███████▊ | 778/999 [00:40<00:11, 19.01it/s]
78%|███████▊ | 780/999 [00:40<00:11, 18.76it/s]
78%|███████▊ | 782/999 [00:41<00:11, 18.79it/s]
78%|███████▊ | 784/999 [00:41<00:11, 18.84it/s]
79%|███████▉ | 787/999 [00:41<00:10, 20.18it/s]
79%|███████▉ | 790/999 [00:41<00:11, 17.80it/s]
79%|███████▉ | 792/999 [00:41<00:11, 17.66it/s]
79%|███████▉ | 794/999 [00:41<00:11, 17.66it/s]
80%|███████▉ | 796/999 [00:41<00:11, 17.48it/s]
80%|███████▉ | 798/999 [00:41<00:11, 17.48it/s]
80%|████████ | 800/999 [00:42<00:11, 17.87it/s]
80%|████████ | 802/999 [00:42<00:11, 16.97it/s]
80%|████████ | 804/999 [00:42<00:11, 17.58it/s]
81%|████████ | 806/999 [00:42<00:10, 17.75it/s]
81%|████████ | 808/999 [00:42<00:10, 17.77it/s]
81%|████████ | 811/999 [00:42<00:09, 20.25it/s]
81%|████████▏ | 814/999 [00:42<00:09, 19.79it/s]
82%|████████▏ | 816/999 [00:42<00:09, 18.61it/s]
82%|████████▏ | 818/999 [00:43<00:10, 17.12it/s]
82%|████████▏ | 820/999 [00:43<00:10, 16.91it/s]
82%|████████▏ | 822/999 [00:43<00:10, 17.04it/s]
82%|████████▏ | 824/999 [00:43<00:10, 17.08it/s]
83%|████████▎ | 826/999 [00:43<00:09, 17.68it/s]
83%|████████▎ | 828/999 [00:43<00:09, 18.05it/s]
83%|████████▎ | 830/999 [00:43<00:09, 18.54it/s]
83%|████████▎ | 832/999 [00:43<00:08, 18.75it/s]
83%|████████▎ | 834/999 [00:43<00:08, 18.92it/s]
84%|████████▎ | 836/999 [00:44<00:09, 16.45it/s]
84%|████████▍ | 838/999 [00:44<00:09, 16.93it/s]
84%|████████▍ | 840/999 [00:44<00:09, 17.61it/s]
84%|████████▍ | 842/999 [00:44<00:08, 18.14it/s]
84%|████████▍ | 844/999 [00:44<00:08, 18.16it/s]
85%|████████▍ | 847/999 [00:44<00:07, 19.52it/s]
85%|████████▌ | 850/999 [00:44<00:07, 20.66it/s]
85%|████████▌ | 853/999 [00:45<00:07, 19.23it/s]
86%|████████▌ | 856/999 [00:45<00:07, 19.87it/s]
86%|████████▌ | 859/999 [00:45<00:06, 20.42it/s]
86%|████████▋ | 862/999 [00:45<00:07, 19.52it/s]
86%|████████▋ | 864/999 [00:45<00:07, 18.93it/s]
87%|████████▋ | 866/999 [00:45<00:07, 18.88it/s]
87%|████████▋ | 868/999 [00:45<00:07, 18.55it/s]
87%|████████▋ | 870/999 [00:45<00:06, 18.90it/s]
87%|████████▋ | 873/999 [00:46<00:06, 20.21it/s]
88%|████████▊ | 876/999 [00:46<00:05, 21.15it/s]
88%|████████▊ | 879/999 [00:46<00:05, 20.42it/s]
88%|████████▊ | 882/999 [00:46<00:06, 18.87it/s]
88%|████████▊ | 884/999 [00:46<00:06, 18.43it/s]
89%|████████▊ | 886/999 [00:46<00:06, 18.32it/s]
89%|████████▉ | 889/999 [00:46<00:05, 19.48it/s]
89%|████████▉ | 892/999 [00:46<00:05, 20.01it/s]
90%|████████▉ | 895/999 [00:47<00:05, 19.83it/s]
90%|████████▉ | 897/999 [00:47<00:05, 19.68it/s]
90%|████████▉ | 899/999 [00:47<00:05, 19.19it/s]
90%|█████████ | 901/999 [00:47<00:05, 18.87it/s]
90%|█████████ | 903/999 [00:47<00:05, 18.48it/s]
91%|█████████ | 906/999 [00:47<00:04, 20.06it/s]
91%|█████████ | 909/999 [00:47<00:04, 19.94it/s]
91%|█████████ | 911/999 [00:47<00:04, 19.62it/s]
91%|█████████▏| 913/999 [00:48<00:04, 19.00it/s]
92%|█████████▏| 915/999 [00:48<00:04, 18.28it/s]
92%|█████████▏| 917/999 [00:48<00:04, 17.71it/s]
92%|█████████▏| 919/999 [00:48<00:04, 18.03it/s]
92%|█████████▏| 922/999 [00:48<00:03, 19.76it/s]
93%|█████████▎| 925/999 [00:48<00:03, 19.77it/s]
93%|█████████▎| 927/999 [00:48<00:03, 19.18it/s]
93%|█████████▎| 929/999 [00:48<00:03, 19.07it/s]
93%|█████████▎| 931/999 [00:49<00:03, 18.80it/s]
93%|█████████▎| 934/999 [00:49<00:03, 19.99it/s]
94%|█████████▎| 936/999 [00:49<00:03, 19.67it/s]
94%|█████████▍| 938/999 [00:49<00:03, 19.08it/s]
94%|█████████▍| 940/999 [00:49<00:03, 19.32it/s]
94%|█████████▍| 943/999 [00:49<00:02, 21.14it/s]
95%|█████████▍| 946/999 [00:49<00:02, 19.30it/s]
95%|█████████▍| 948/999 [00:49<00:02, 19.31it/s]
95%|█████████▌| 950/999 [00:50<00:02, 19.22it/s]
95%|█████████▌| 952/999 [00:50<00:02, 19.17it/s]
95%|█████████▌| 954/999 [00:50<00:02, 18.42it/s]
96%|█████████▌| 956/999 [00:50<00:02, 18.37it/s]
96%|█████████▌| 958/999 [00:50<00:02, 18.17it/s]
96%|█████████▌| 960/999 [00:50<00:02, 18.37it/s]
96%|█████████▋| 963/999 [00:50<00:01, 19.35it/s]
97%|█████████▋| 965/999 [00:50<00:01, 19.44it/s]
97%|█████████▋| 967/999 [00:50<00:01, 19.50it/s]
97%|█████████▋| 969/999 [00:51<00:01, 18.48it/s]
97%|█████████▋| 971/999 [00:51<00:01, 18.73it/s]
97%|█████████▋| 973/999 [00:51<00:01, 18.67it/s]
98%|█████████▊| 975/999 [00:51<00:01, 19.04it/s]
98%|█████████▊| 977/999 [00:51<00:01, 19.15it/s]
98%|█████████▊| 979/999 [00:51<00:01, 19.01it/s]
98%|█████████▊| 982/999 [00:51<00:00, 20.48it/s]
99%|█████████▊| 985/999 [00:51<00:00, 19.25it/s]
99%|█████████▉| 987/999 [00:51<00:00, 18.95it/s]
99%|█████████▉| 989/999 [00:52<00:00, 18.56it/s]
99%|█████████▉| 991/999 [00:52<00:00, 18.45it/s]
99%|█████████▉| 993/999 [00:52<00:00, 18.26it/s]
100%|█████████▉| 995/999 [00:52<00:00, 18.00it/s]
100%|█████████▉| 997/999 [00:52<00:00, 17.82it/s]
100%|██████████| 999/999 [00:52<00:00, 17.95it/s]
100%|██████████| 999/999 [00:52<00:00, 18.98it/s]
plt.plot(times, positions[:,2])
plt.axhline(y=2.25, color='red', linestyle='-')
plt.axhline(y=1.75, color='red', linestyle='-')
plt.show()