Add histogram-making functions and plotting scripts

This commit is contained in:
J.-S. Caux
2020-05-15 14:55:21 +02:00
parent 6ffcf1a98c
commit 15e9fdb5d4
5 changed files with 236 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
import argparse
import plotly.graph_objects as go
import numpy
parser = argparse.ArgumentParser()
parser.add_argument('omegafile', help='Omega filename')
parser.add_argument('dsffile', help='DSF filename')
args = parser.parse_args()
omega = numpy.loadtxt(args.omegafile)
dsf = numpy.loadtxt(args.dsffile)
x = [o for o in omega]
y = [d for d in dsf]
fig = go.Figure(data=go.Scatter(x=x, y=y))
fig.update_layout(title=args.dsffile.rpartition('/')[2])
fig.show()
+21
View File
@@ -0,0 +1,21 @@
import argparse
import plotly.graph_objects as go
import numpy
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*')
args = parser.parse_args()
print(args)
fig = go.Figure()
for datafilename in args.filenames:
data = numpy.loadtxt(datafilename, delimiter="\t", usecols=[0,1,2,3])
x = [line[0] for line in data]
y = [line[3] for line in data]
fig.add_trace(go.Bar(x=x, y=y))
fig.show()
+38
View File
@@ -0,0 +1,38 @@
import argparse
import plotly.graph_objects as go
import numpy
parser = argparse.ArgumentParser()
parser.add_argument('filename', help='Enter the sliced data filename')
args = parser.parse_args()
data = numpy.loadtxt(args.filename, delimiter="\t")
# The data is in row format
x = [i for i in range(len(data[0]))]
fig = go.Figure()
for line in data:
fig.add_trace(go.Bar(x=x, y=[d for d in line]))
steps = []
for i in range(len(fig.data)):
step = dict(
method="update",
args=[{'visible': [False] * len(fig.data)},
{'title': "Slider switched to step: " + str(i)}],)
step["args"][0]["visible"][i] = True
steps.append(step)
sliders = [dict(
active=0,
currentvalue={"prefix": "bin box "},
pad={"t": 50},
steps=steps)]
fig.update_yaxes(range=[0,0.2])
fig.update_layout(
sliders=sliders
)
fig.show()