resolve FutureWarning for pandas 2 with ser.iloc[pos]

unaschneck · Nov 6, 2023 · f8385fd · f8385fd
1 parent f73ec48
commit f8385fd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/pydar/error_handling.py b/pydar/error_handling.py
@@ -108,8 +108,8 @@ def errorHandlingConvertFlybyIDToObservationNumber(flyby_id=None):
 	valid_flyby_ids = []
 	flyby_id_found = False
 	for index, row in flyby_dataframe.iterrows():
-		valid_flyby_ids.append(row[0])
-		if row[0] == flyby_id:
+		valid_flyby_ids.append(row.iloc[0])
+		if row.iloc[0] == flyby_id:
 			flyby_id_found = True
 			break
 	if not flyby_id_found:
@@ -128,7 +128,7 @@ def errorHandlingConvertObservationNumberToFlybyID(flyby_observation_num=None):
 	valid_observation_nums = []
 	observation_num_found = False
 	for index, row in flyby_dataframe.iterrows():
-		take_ob_num = "0" + row[1].split(" ")[1]
+		take_ob_num = "0" + row.iloc[1].split(" ")[1]
 		valid_observation_nums.append(take_ob_num)
 		if take_ob_num == flyby_observation_num:
 			observation_num_found = True

diff --git a/pydar/extract_flyby_parameters.py b/pydar/extract_flyby_parameters.py
@@ -46,8 +46,8 @@ def convertFlybyIDToObservationNumber(flyby_id=None):
 	flyby_csv_file = os.path.join(os.path.dirname(__file__), 'data', 'cassini_flyby.csv')  # get file's directory, up one level, /data/*.csv
 	flyby_dataframe = pd.read_csv(flyby_csv_file)
 	for index, row in flyby_dataframe.iterrows():
-		if row[0] == flyby_id:
-			observation_number = row[1].split(" ")[1]
+		if row.iloc[0] == flyby_id:
+			observation_number = row.iloc[1].split(" ")[1]
 			while len(observation_number) < 4:
 				observation_number = "0" + observation_number # set all radar take numbers to be four digits long: 229 -> 0229
 			return observation_number
@@ -66,9 +66,9 @@ def convertObservationNumberToFlybyID(flyby_observation_num=None):
 	flyby_csv_file = os.path.join(os.path.dirname(__file__), 'data', 'cassini_flyby.csv')  # get file's directory, up one level, /data/*.csv
 	flyby_dataframe = pd.read_csv(flyby_csv_file)
 	for index, row in flyby_dataframe.iterrows():
-		take_ob_num = "0" + row[1].split(" ")[1]
+		take_ob_num = "0" + row.iloc[1].split(" ")[1]
 		if take_ob_num == flyby_observation_num:
-			return row[0] # returns flyby ID
+			return row.iloc[0] # returns flyby ID
 
 def retrieveJPLCoradrOptions():
 	# Read JPL Options from CSV