From 2fa7ecd9b243c1ee664073a8b3d1c47a75c9da7d Mon Sep 17 00:00:00 2001 From: guanjihuan Date: Sun, 6 Feb 2022 18:35:05 +0800 Subject: [PATCH] 0.0.57 --- API_Reference/API_Reference.py | 5 +++++ PyPI/setup.cfg | 2 +- PyPI/src/guan/__init__.py | 33 +++++++++++++++++++++++++++++---- 3 files changed, 35 insertions(+), 5 deletions(-) diff --git a/API_Reference/API_Reference.py b/API_Reference/API_Reference.py index 4d11d20..f6f0339 100644 --- a/API_Reference/API_Reference.py +++ b/API_Reference/API_Reference.py @@ -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') +# preprocess + +parameter_array = guan.preprocess_for_parallel_calculations(parameter_array_all, cpus=1, task_index=0) + + # plot figures diff --git a/PyPI/setup.cfg b/PyPI/setup.cfg index e959e58..cbe01d6 100644 --- a/PyPI/setup.cfg +++ b/PyPI/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = guan -version = 0.0.54 +version = 0.0.57 author = guanjihuan author_email = guanjihuan@163.com description = An open source python package diff --git a/PyPI/src/guan/__init__.py b/PyPI/src/guan/__init__.py index fa7342a..f8327a4 100644 --- a/PyPI/src/guan/__init__.py +++ b/PyPI/src/guan/__init__.py @@ -12,8 +12,9 @@ # # Module 8: quantum_transport # # Module 9: topological_invariant # # Module 10: read_and_write -# # Module 11: plot_figures -# # Module 12: others +# # Module 11: preprocess +# # Module 12: plot_figures +# # Module 13: others # 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): 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