Skip to content

Commit 2c90e27

Browse files
committed
A bit modification
1 parent da10ee8 commit 2c90e27

2 files changed

Lines changed: 141 additions & 80 deletions

File tree

Calculation/example/calculation_accuracy_1d_sympy.py

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,13 +65,36 @@
6565
H2 += tmp0 + 1/2 * sp.eye(Nsite)
6666
# H2 += Vs/2 * sp.eye(Nsite)
6767

68+
q_array = np.linspace(-1, 1, 11)
69+
x = sp.symbols('x')
70+
71+
72+
#import numpy
73+
#a = numpy.arange(100)
74+
#expr = sin(x)**2 + cos(x)**3
75+
#display(expr)
76+
#f = lambdify(x, expr, "numpy") # numpyを使わないと遅い
77+
#f(a)
78+
79+
eqH2_array = sp.lambdify(q, H2)
80+
print(q_array)
81+
print(eqH2_array(np.array([1, 2])))
82+
83+
for eqH2 in eqH2_array(q_array):
84+
start = time.time()
85+
print(eqH2.shape)
86+
# E0, P = np.linalg.eig(eqH2)
87+
# rearrangedEvalsVecs = sorted(zip(E0, P.T), key=lambda x: x[0].real, reverse=False)
88+
# E = eqH2.eigenvals()
89+
elapsed_time = time.time() - start
90+
91+
print(elapsed_time)
6892
# print(tmp0)
6993
# for i in range(Nsite):
7094
# H[i][i] = G**2 * (2*(i-Nsite/2) + q[i_q])**2 + s/2
7195

72-
A = sp.Matrix(([3, 2], [-2, 3]))
7396
print(H2)
74-
print(H2.eigenvals())
97+
#print(H2.eigenvals())
7598

7699
# for m_tmp in tqdm(m_list):
77100
# # for m_tmp in m_list:

0 commit comments

Comments
 (0)