コード例 #1
0
args = parser.parse_args()
nstars = int(args.nstars)
nbins = int(args.nbins)
scale = bool(args.scale)

colors = plt.rcParams['axes.prop_cycle'].by_key()['color'][:7]

filenames = glob('/media/connor/Lore/ultramag/steve_5899_v2/star_0*.txt')
stars = []

print('\nLoading stars...')
pb = ProgressBar(len(filenames[:nstars]))
for i, filename in enumerate(filenames[:nstars]):
    star = Star(filename)
    if star.data is not None:
        star.filter()
        stars.append(star)
    pb.update()

fig, ax1 = plt.subplots(1, 1, figsize=[10, 10])

print('\nSplitting data into nights and plotting...')
pb = ProgressBar(len(stars))

all_times = []
all_mags = []

for i, star in enumerate(stars):
    nights = star.split_nights()
    c = colors[i % len(colors)]