plot_scheme.py

import numpy as np
import matplotlib.pyplot as plt
import csv


def plot_wavenumber(k,mkr,mki,formula):

	equation = r'$'+formula+'$'
	plt.title(equation)

	plt.plot(k,k, color="grey", linewidth=0.5, linestyle="-", label="exact")
	plt.plot(k,mkr, color="red", linewidth=1.5, linestyle="-", label="real")
	plt.plot(k,mki, color="blue", linewidth=1.5, linestyle="-", label="image")
	plt.legend()
	plt.xlabel('wavenumber')
	plt.xlim(0, np.pi)
	plt.ylabel('modified wavenumber')
	plt.ylim(min(mki)*1.01, np.pi)
	plt.axhline(linewidth=0.5, color='black')
	plt.show()
def file_write_wavenumber(filename,k,mkr,mki):
	with open(filename, "w") as f:
		print("{:>18}  {:>18}  {:>18}".format('k','real(modified_k)','image(modified_k)'),file=f)
		for n in range(0,len(k)):
			print("{:.12E}  {:.12E}  {:.12E}".format(k[n],mkr[n],mki[n]),file=f)
	print(' << ',filename)


def plot_wavenumber_filter(k,mkr,mki,formula):

	equation = r'$'+formula+'$'
	plt.title(equation)

	plt.text(0.05, 1.1, r'$\alpha = 0.49$', fontsize=14)  

	plt.plot(k,mkr, color="red", linewidth=1.5, linestyle="-", label="real")
	plt.plot(k,mki, color="blue", linewidth=1.5, linestyle="-", label="image")
	plt.legend()
	plt.xlabel('wavenumber')
	plt.xlim(0, np.pi)
	plt.ylabel('spectral function')
	plt.ylim(-0.2, 1.2)
	plt.axhline(linewidth=0.5, color='black')
	plt.show()