-
Notifications
You must be signed in to change notification settings - Fork 0
/
Eigenvector_Loader_Z.py
61 lines (39 loc) · 1.03 KB
/
Eigenvector_Loader_Z.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import numpy as np
import matplotlib.pyplot as plt
XMIN = -1.0
XMAX = 1.0
XDIV = 23
YMIN = -1.0
YMAX = 1.0
YDIV = 23
ZMIN = 3.32
ZMAX = 5.32
ZDIV = 23
XSTEPSIZE = (XMAX-XMIN)/(XDIV-1)
ZArray = np.array([])
s = int((XMAX/XSTEPSIZE))
B = np.array([])
xgrid = np.linspace(XMIN, XMAX, XDIV)
ygrid = np.linspace(YMIN, YMAX, YDIV)
zgrid = np.linspace(ZMIN, ZMAX, ZDIV)
meshx, meshy, meshz = np.meshgrid(xgrid, ygrid, zgrid, sparse=False, indexing="xy")
A=(np.load("23by231point0evec.npy"))
print(A)
for val in A:
string_val = str(val)
parentremover = string_val.split('[')
keep = parentremover[1]
secondparentremover = keep.split(']')
nextkeep = secondparentremover[0]
realimg = nextkeep.split('+0.j')
realpart= float(realimg[0])
scientific_notation = "{:.4e}".format(realpart)
final = float(scientific_notation)
B = np.append(B, final)
B = np.reshape(B, (XDIV, YDIV, ZDIV))
print(B)
ZArray = B[s:s+1,s:s+1]
ZArray = np.resize(ZArray,(ZDIV))
print(ZArray)
plt.plot(zgrid, ZArray)
plt.show()