from params import avoid_table, marana_yoga_table from params import NAKSHATRAS import numpy as np user_nakshatra = NAKSHATRAS.MAGAM num_days = 90 mins_step = 30 a = Almanac() t0 = a.time_scale.utc(2020, 4, 1) t1 = a.time_scale.utc(2020, 4, 1, 0, 24 * 60 * 366) month_times, months = a._get_month_start_end(t0, t1) # sun_rise, sun_set = a._compute_sun_rise_sun_set(t0, t1) tframe = a.time_scale.utc(2020, 4, 1, 0, range(0, 24 * 60 * num_days, mins_step)) a.time = tframe a.compute_almanac() mask = np.ones(tframe.shape, dtype=np.bool) print(f'started with {mask.sum()}') for t in avoid_thitis_list: mask[(t == a.thithi.name)] = False print(f'after removing thithis {avoid_thitis_list} -- {mask.sum()}') for n in nakshaktra_ignore_list: mask[n == a.nakshaktra.name] = False print(f'after removing nakshaktras {nakshaktra_ignore_list} -- {mask.sum()}') for day, thithis in avoid_thithi_on_days.items(): cur_days = (a.vara == day) for t in thithis: mask[(t == a.thithi.name) & cur_days] = False print(f'after removing thithi on days {avoid_thithi_on_days} -- {mask.sum()}') for day, nakshaktras in avoid_nakshaktras_on_day.items():