ester/postprocessing/plot.py at master · ester-project/ester · GitHub

27 lines (24 loc) · 596 Bytes

# visualize a chosen quantity for each co-latitudes theta

# pass model file name and quantity as parameters

from ester import *

import matplotlib.pyplot as plt

import numpy as np

import argparse

parser = argparse.ArgumentParser()

parser.add_argument(

"model",

type=str,

help="the path to the folder containing ESTER model",

)

parser.add_argument(

"--plot",

default="rho",

type=str,

help=f"which parameter to plot"

)

ARGS = parser.parse_args()

a=star2d(ARGS.model)

plt.plot(a.r[:],np.log10(getattr(a, ARGS.plot)[:]))

plt.xlabel("r")

plt.ylabel(ARGS.plot)

plt.show()

Expand file tree