Add files via upload

Author: Evolutionary-Intelligence

tutorials/plot_3d_landscape_for_rotated_functions.py

@@ -0,0 +1,37 @@
+"""This script has been used in Qiqi Duan's Ph.D. Dissertation (HIT&SUSTech).
+
+    Chinese: 该绘图脚本被段琦琦的博士论文（哈工大与南科大联合培养）所使用。
+"""
+import matplotlib
+import matplotlib.pyplot as plt
+
+from pypop7.benchmarks.rotated_functions import ellipsoid
+from pypop7.benchmarks.rotated_functions import generate_rotation_matrix
+from pypop7.benchmarks.utils import generate_xyz
+
+
+matplotlib.rcParams['font.family'] = 'sans-serif'
+matplotlib.rcParams['font.sans-serif'] = 'SimSun'
+matplotlib.rcParams['axes.unicode_minus'] = False
+matplotlib.rcParams['font.size'] = 10  # 对应5号字体
+
+
+if __name__ == '__main__':
+    generate_rotation_matrix(ellipsoid, 2, 72)
+    bound = [-10.0, 10.0]
+    fig = plt.figure(figsize=(2.5, 2.5))
+    ax = fig.add_subplot(1, 1, 1, projection='3d')
+    x, y, z = generate_xyz(ellipsoid, bound, bound, 500)
+    ax.plot_surface(x, y, z, cmap='cool',
+                    rstride=100, cstride=100, alpha=0.7)
+    ax.set_title('病态函数', fontsize=10)
+    ax.set_xlim(bound)
+    ax.set_ylim(bound)
+    ax.set_xticks([-10.0, -5.0, 0.0, 5.0, 10.0], fontsize=10)
+    ax.set_yticks([-10.0, -5.0, 0.0, 5.0, 10.0], fontsize=10)
+    ax.set_zticks([0.0e8, 0.5e8, 1.5e8], ['0.0e8', '0.5e8', '1.5e8'], fontsize=10)
+    ax.set_xlabel('维度1', fontsize=10)
+    ax.set_ylabel('维度2', fontsize=10)
+    ax.set_zlabel('适应值', fontsize=10, rotation='vertical')
+    plt.savefig('3d_ellipsoid.png', dpi=700, bbox_inches='tight')
+    plt.show()