FYS3150/Project-5
3
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity

Clean up script

Browse Source
This commit is contained in:
Janita Willumsen 2024-01-01 15:38:03 +01:00
parent 70057de961
commit 18aa574624
1 changed files with 5 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
python_scripts/colormap_all.py
View File

@ -30,74 +30,31 @@ def plot():
lines = f.readlines() lines = f.readlines()
size = int(lines[0]) size = int(lines[0])
for i, line in enumerate(lines[1:]): for i, line in enumerate(lines[1:]):
# Create figures for each plot
# fig1, ax1 = plt.subplots()
# fig2, ax2 = plt.subplots()
# fig3, ax3 = plt.subplots()
arr = line.strip().split("\t") arr = line.strip().split("\t")
arr = np.asarray(list(map(lambda x: complex(*ast.literal_eval(x)), arr))) arr = np.asarray(list(map(lambda x: complex(*ast.literal_eval(x)), arr)))
# Reshape and transpose array # Reshape and transpose array
arr = arr.reshape(size, size).T arr = arr.reshape(size, size).T
# Plot color maps # Plot color maps
color_map1 = axes[0, i].imshow( axes[0, i].imshow(
np.multiply(arr, arr.conj()).real, np.multiply(arr, arr.conj()).real,
interpolation="nearest", interpolation="nearest",
cmap=sns.color_palette("mako", as_cmap=True) cmap=sns.color_palette("mako", as_cmap=True)
) )
color_map2 = axes[1, i].imshow( axes[1, i].imshow(
arr.real, arr.real,
interpolation="nearest", interpolation="nearest",
cmap=sns.color_palette("mako", as_cmap=True) cmap=sns.color_palette("mako", as_cmap=True)
) )
color_map3 = axes[2, i].imshow( axes[2, i].imshow(
arr.imag, arr.imag,
interpolation="nearest", interpolation="nearest",
cmap=sns.color_palette("mako", as_cmap=True) cmap=sns.color_palette("mako", as_cmap=True)
) )
# Create color bar # Create tight subplots and save figure
fig.subplots_adjust(wspace=0, hspace=0) fig.subplots_adjust(wspace=0, hspace=0)
# fig.colorbar(color_map1, ax=axes[0, 2]) fig.savefig(f"latex/images/color_map_all.pdf", bbox_inches="tight")
# fig.colorbar(color_map2, ax=axes[1, 2])
# fig.colorbar(color_map3, ax=axes[2, 2])
# Remove grids
# for i in range(3):
# axes[i, 0].set_xticks([])
# axes[i, 0].set_yticks([])
# axes[i, 0].
# axes[i, 1].set_xticks([])
# axes[i, 1].set_yticks([])
# axes[i, 2].set_xticks([])
# axes[i, 2].set_yticks([])
# axes[0].set_xticks([])
# axes[0].set_yticks([])
# axes[1].set_xticks([])
# axes[].set_yticks([])
# axes[].set_xticks([])
# axes[].set_yticks([])
# # Set labels
# ax1.set_xlabel("x-axis")
# ax1.set_ylabel("y-axis")
# ax2.set_xlabel("x-axis")
# ax2.set_ylabel("y-axis")
# ax3.set_xlabel("x-axis")
# ax3.set_ylabel("y-axis")
# # Save the figures
plt.savefig(f"latex/images/color_map_all.pdf", bbox_inches="tight")
# fig2.savefig(f"latex/images/color_map_{i}_real.pdf", bbox_inches="tight")
# fig3.savefig(f"latex/images/color_map_{i}_imag.pdf", bbox_inches="tight")
# # Close figures
# plt.close(fig1)
# plt.close(fig2)
# plt.close(fig3)
if __name__ == "__main__": if __name__ == "__main__":