1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # from SigMod import *
# import matplotlib.pyplot as plt
from GolemShot import GolemShot
import numpy as np
def main():
shot = GolemShot(0)
time, dz = shot.getDz()
np.savetxt('plasma_position.txt', np.array((time, dz)).T, fmt='%f')
return
# time -= shot['plasma_start']
# plt.plot(time, dz, label='dz_mirnov')
# plt.plot(time, np.zeros(shape=len(dz)), label='dz')
# plt.title("vertical displacement")
# plt.legend()
# plt.show()
if __name__ == '__main__':
main()
|