Wiele funkcji biblioteki ASE oczekuje danych w postaci trajektorii. Poniższy krótki kod przekształca surowe dane otrzymane z symulacji w zestaw trajektorii przydatnych np. przy wizualizacji.
In [1]:
import ase.io
import pickle
import gzip
import nglview
In [2]:
# Wczytanie danych wyliczonych przez program VASP przygotowanych wczesniej
md={}
for k,tr in pickle.load(gzip.open('data/md_PtFePt.p.gz','rb')).items():
T=int(k.split('/')[-1][1:])
md[T]=tr
print(sorted(md.keys()))
In [3]:
for T in md:
ase.io.write('data/md_T_%d.traj' % (T,), md[T][1])
In [ ]: