Simple plot of 1D data
The example below demonstrates how to plot 1D data, precisely:
datadim 1 # dimension of block size: 1=nx, 2=nx*ny, 3=nx*ny*nz
The simple plotting script may look like:
#!/usr/bin/python3
# Before the run:
# Get wdata lib:
# pip3 install wdata
# (add -U if you want to update)
# For more info see:
# https://pypi.org/project/wdata/
import numpy as np
import matplotlib.pyplot as plt
from wdata.io import WData, Var
# I read data directly from dwarf,
# Before running the script I mounted file system:
# sshfs gabrielw@dwarf.if.pw.edu.pl:/home2/ /home/gabrielw/home2-dwarf/
fwtxt="/home/gabrielw/home2-dwarf/scratch/barrus-gw/h1d/lambda006.wtxt"
data = WData.load(fwtxt)
fig, ax = plt.subplots()
ax.plot(data.xyz[0], data.rho_a[-1]) # plot last frame [-1]
ax.set(xlabel='x', ylabel=r'$n_{\uparrow}(x)$')
ax.grid()
fig.savefig("rho_a.png")
plt.show()
Simple plot of 2D data
The example below demonstrates how to plot 1D data, precisely:
datadim 2 # dimension of block size: 1=nx, 2=nx*ny, 3=nx*ny*nz
The simple plotting script may look like:
#!/usr/bin/python3
# Before run:
# Get wata lib:
# pip3 install wdata
# (add -U if you want to update)
# For more info see:
# https://pypi.org/project/wdata/
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from wdata.io import WData, Var
# I read data directly from dwarf,
# Before running the script I mounted file system:
# sshfs gabrielw@dwarf.if.pw.edu.pl:/home2/ /home/gabrielw/home2-dwarf/
fwtxt="/home/gabrielw/home2-dwarf/archive/ntg/repository/atomicjj/CALC-BCS/td-round5-bcs/td1-bcs-z0.02-V0.6.wtxt"
data = WData.load(fwtxt)
ratio=data.Nxyz[0]/data.Nxyz[1]
scale=2.0
fig, ax = plt.subplots(figsize=(scale*ratio,1.2*scale))
im = ax.imshow(np.transpose(data.rho_a[0]), interpolation='bilinear', extent=[0, data.Nxyz[0], 0, data.Nxyz[1]], aspect='auto')
fig.colorbar(im, ax=ax, orientation="horizontal")
plt.tight_layout()
fig.savefig("rho_a.png")
plt.show()