diff --git a/star_chart_spherical_projection/generate_star_chart.py b/star_chart_spherical_projection/generate_star_chart.py index 9b7d122..38131c5 100644 --- a/star_chart_spherical_projection/generate_star_chart.py +++ b/star_chart_spherical_projection/generate_star_chart.py @@ -28,15 +28,15 @@ southern_declination_max = -90 # Start Year (JP2000) j2000 = 2000 # start year of the star catalogue (jan 1 2000 via IAU) -def _get_stars(selectStars=[]): +def _get_stars(select_stars=[]): # selectStars only returns a subset of all the stars saved, empty will return all in the star_data.csv file # stars: ["name", "RA: HH.MM.SS", Declination DD.SS, Proper Motion Speed (mas/yr), Proper Motion Angle (DD.SS), Magnitude (V, Visual)] star_data_list = [] star_csv_file = os.path.join(os.path.dirname(__file__), 'data', 'stars_with_data.csv') # get file's directory, up one level, /data/4_all_stars_data.csv star_dataframe = pd.read_csv(star_csv_file) for index, row in star_dataframe.iterrows(): - if len(selectStars) > 0: # get only a subset of all stars - if row["Common Name"] in selectStars: + if len(select_stars) > 0: # get only a subset of all stars + if row["Common Name"] in select_stars: star_data_list.append(row.tolist()) else: star_data_list.append(row.tolist())