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

0.0.57

Browse Source
This commit is contained in:
guanjihuan 2022-02-06 18:35:05 +08:00
parent 5a4ecd2926
commit 2fa7ecd9b2
3 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
API_Reference/API_Reference.py
View File

@ -231,6 +231,11 @@ guan.write_one_dimensional_data(x_array, y_array, filename='a', format='txt')
guan.write_two_dimensional_data(x_array, y_array, matrix, filename='a', format='txt') guan.write_two_dimensional_data(x_array, y_array, matrix, filename='a', format='txt')
# preprocess
parameter_array = guan.preprocess_for_parallel_calculations(parameter_array_all, cpus=1, task_index=0)
# plot figures # plot figures

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.54 version = 0.0.57
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

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

@ -12,8 +12,9 @@
# # Module 8: quantum_transport # # Module 8: quantum_transport
# # Module 9: topological_invariant # # Module 9: topological_invariant
# # Module 10: read_and_write # # Module 10: read_and_write
# # Module 11: plot_figures # # Module 11: preprocess
# # Module 12: others # # Module 12: plot_figures
# # Module 13: others
# import packages # import packages
@ -1442,7 +1443,31 @@ def write_two_dimensional_data(x_array, y_array, matrix, filename='a', format='t
# Module 11: plot figures # Module 11: preprocess
def preprocess_for_parallel_calculations(parameter_array_all, cpus=1, task_index=0):
num_all = np.array(parameter_array_all).shape[0]
if num_all%cpus == 0:
num_parameter = int(num_all/cpus)
parameter_array = parameter_array_all[task_index*num_parameter:(task_index+1)*num_parameter]
else:
num_parameter = int(num_all/(cpus-1))
if task_index != cpus-1:
parameter_array = parameter_array_all[task_index*num_parameter:(task_index+1)*num_parameter]
else:
parameter_array = parameter_array_all[task_index*num_parameter:num_all]
return parameter_array
# Module 12: plot figures
def plot(x_array, y_array, xlabel='x', ylabel='y', title='', show=1, save=0, filename='a', format='jpg', dpi=300, type='', y_min=None, y_max=None, linewidth=None, markersize=None): def plot(x_array, y_array, xlabel='x', ylabel='y', title='', show=1, save=0, filename='a', format='jpg', dpi=300, type='', y_min=None, y_max=None, linewidth=None, markersize=None):
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -1539,7 +1564,7 @@ def plot_contour(x_array, y_array, matrix, xlabel='x', ylabel='y', title='', sho
# Module 12: others # Module 13: others
## download ## download