Fix issue #30

- Clip reference year to current year when calculating cit/year to avoid zero division in cit/year
WittmannF · Nov 15, 2023 · 05dc8b7 · 05dc8b7
1 parent fa67ad8
commit 05dc8b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sortgs.py b/sortgs.py
@@ -280,8 +280,8 @@ def main():
                         columns=['Author', 'Title', 'Citations', 'Year', 'Publisher', 'Venue', 'Source'])
     data.index.name = 'Rank'
 
-    # Add columns with number of citations per year
-    data['cit/year']=data['Citations']/(end_year + 1 - data['Year'])
+    # Avoid years that are higher than the current year by clipping it to end_year
+    data['cit/year']=data['Citations']/(end_year + 1 - data['Year'].clip(upper=end_year))
     data['cit/year']=data['cit/year'].round(0).astype(int)
 
     # Sort by the selected columns, if exists