update

2020.08.29_several_python_functions/2D_data_plot_and_write/plot_with_2D_data.py

@@ -0,0 +1,30 @@
+import numpy as np
+from math import *  
+# import os
+# os.chdir('D:/data')  # 设置路径
+
+
+def main():
+    k1 = np.arange(-pi, pi, 0.05)
+    k2 = np.arange(-pi, pi, 0.05)
+    value = np.ones((k2.shape[0], k1.shape[0]))
+    plot_matrix(k1, k2, value)
+
+
+def plot_matrix(k1, k2, matrix):
+    import matplotlib.pyplot as plt
+    from mpl_toolkits.mplot3d import Axes3D
+    from matplotlib import cm
+    from matplotlib.ticker import LinearLocator, FormatStrFormatter
+    fig = plt.figure()
+    ax = fig.gca(projection='3d')
+    k1, k2 = np.meshgrid(k1, k2)
+    ax.plot_surface(k1, k2, matrix, cmap=cm.coolwarm, linewidth=0, antialiased=False) 
+    plt.xlabel('k1')
+    plt.ylabel('k2') 
+    ax.set_zlabel('Z')  
+    plt.show()
+
+
+if __name__ == '__main__':
+    main()

2020.08.29_several_python_functions/2D_data_plot_and_write/write_2D_data_into_txt_file.py

@@ -0,0 +1,31 @@
+import numpy as np
+from math import *  
+# import os
+# os.chdir('D:/data')  # 设置路径
+
+
+def main():
+    k1 = np.arange(-pi, pi, 0.05)
+    k2 = np.arange(-pi, pi, 0.05)
+    value = np.ones((k2.shape[0], k1.shape[0]))
+    write_matrix(k1, k2, value)
+
+
+def write_matrix(k1, k2, matrix):
+    with open('a.txt', 'w') as f:
+        # np.set_printoptions(suppress=True)  # 取消输出科学记数法
+        f.write('0           ')
+        for k10 in k1:
+            f.write(str(k10)+'   ')
+        f.write('\n')
+        i0 = 0
+        for k20 in k2:
+            f.write(str(k20))
+            for j0 in range(k1.shape[0]):
+                f.write('  '+str(matrix[i0, j0])+'   ')
+            f.write('\n')
+            i0 += 1   
+
+
+if __name__ == '__main__':
+    main()