37 行
673 B
Python
可执行文件
37 行
673 B
Python
可执行文件
#! /usr/bin/env python
|
|
|
|
"""
|
|
Plot fixed momentum DSF.
|
|
|
|
Usage: python plot_dsf_k_fixed.py [omega file] [dsf file]
|
|
"""
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import sys
|
|
|
|
omegafile = str(sys.argv[1])
|
|
dsffile = str(sys.argv[2])
|
|
|
|
omega = np.loadtxt(omegafile)
|
|
dsf = np.loadtxt(dsffile)
|
|
|
|
plt.plot(omega, dsf)
|
|
|
|
plt.xlabel('$\omega$')
|
|
plt.ylabel('$S (k, \omega)$')
|
|
|
|
elements = dsffile.split('_')
|
|
|
|
c_int = elements[4]
|
|
L = elements[6]
|
|
N = elements[8]
|
|
iK = elements[13]
|
|
width = elements[23].partition('.')[0]
|
|
|
|
rho = int(N)/int(L)
|
|
kokF = int(iK)*0.5/int(L)
|
|
plt.title(f'c={c_int}, rho={rho} (N={N}), k={kokF}k_F, w={width}')
|
|
|
|
plt.savefig(dsffile.replace('.', '_') + '.png')
|