guanjihuan/py.guanjihuan.com
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
guanjihuan 2022-02-22 19:15:28 +08:00
parent d65d4c1940
commit f11d5f4abb
3 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
PyPI/setup.cfg
View File

@ -1,7 +1,7 @@
[metadata] [metadata]
# replace with your username: # replace with your username:
name = guan name = guan
version = 0.0.70 version = 0.0.71
author = guanjihuan author = guanjihuan
author_email = guanjihuan@163.com author_email = guanjihuan@163.com
description = An open source python package description = An open source python package

6
PyPI/src/guan/__init__.py
View File

@ -571,6 +571,7 @@ def find_vector_with_the_same_gauge_with_binary_search(vector_target, vector_ref
return vector_target return vector_target
def find_vector_with_fixed_gauge_by_making_one_component_real(vector, precision=0.005, index=None): def find_vector_with_fixed_gauge_by_making_one_component_real(vector, precision=0.005, index=None):
vector = np.array(vector)
if index == None: if index == None:
index = np.argmax(np.abs(vector)) index = np.argmax(np.abs(vector))
sign_pre = np.sign(np.imag(vector[index])) sign_pre = np.sign(np.imag(vector[index]))
@ -585,7 +586,6 @@ def find_vector_with_fixed_gauge_by_making_one_component_real(vector, precision=
return vector return vector
def find_vector_array_with_fixed_gauge_by_making_one_component_real(vector_array, precision=0.005): def find_vector_array_with_fixed_gauge_by_making_one_component_real(vector_array, precision=0.005):
import guan
vector_sum = 0 vector_sum = 0
Num_k = np.array(vector_array).shape[0] Num_k = np.array(vector_array).shape[0]
for i0 in range(Num_k): for i0 in range(Num_k):
@ -596,7 +596,8 @@ def find_vector_array_with_fixed_gauge_by_making_one_component_real(vector_array
return vector_array return vector_array
def rotation_of_degenerate_vectors(vector1, vector2, index1, index2, precision=0.01, criterion=0.01, show_theta=0): def rotation_of_degenerate_vectors(vector1, vector2, index1, index2, precision=0.01, criterion=0.01, show_theta=0):
import cmath vector1 = np.array(vector1)
vector2 = np.array(vector2)
if np.abs(vector1[index2])>criterion or np.abs(vector2[index1])>criterion: if np.abs(vector1[index2])>criterion or np.abs(vector2[index1])>criterion:
for theta in np.arange(0, 2*pi, precision): for theta in np.arange(0, 2*pi, precision):
if show_theta==1: if show_theta==1:
@ -616,7 +617,6 @@ def rotation_of_degenerate_vectors(vector1, vector2, index1, index2, precision=0
return vector1, vector2 return vector1, vector2
def rotation_of_degenerate_vectors_array(vector1_array, vector2_array, precision=0.01, criterion=0.01, show_theta=0): def rotation_of_degenerate_vectors_array(vector1_array, vector2_array, precision=0.01, criterion=0.01, show_theta=0):
import guan
Num_k = np.array(vector1_array).shape[0] Num_k = np.array(vector1_array).shape[0]
vector1_sum = 0 vector1_sum = 0
for i0 in range(Num_k): for i0 in range(Num_k):

22
Tutorial/09_fixed_gauge_of_wave_functions.py Normal file
View File

@ -0,0 +1,22 @@
import numpy as np
import cmath
import guan
# Example 1
vector = np.array([np.sqrt(0.5), np.sqrt(0.5)])*cmath.exp(np.random.uniform(0, 1)*1j)
print('\nExample 1\n', vector)
print(np.dot(vector.transpose().conj(), vector), '\n')
vector = guan.find_vector_with_fixed_gauge_by_making_one_component_real(vector, precision=0.001)
print(vector)
print(np.dot(vector.transpose().conj(), vector), '\n')
# Example 2
vector = np.array([1, 0])*cmath.exp(np.random.uniform(0, 1)*1j)
print('\nExample 2\n', vector)
print(np.dot(vector.transpose().conj(), vector), '\n')
vector = guan.find_vector_with_fixed_gauge_by_making_one_component_real(vector, precision=0.001)
print(vector)
print(np.dot(vector.transpose().conj(), vector), '\n')