refactoring

2023-03-31 12:17:02 +02:00
parent 70e5688be3
commit 0fa873e461
2 changed files with 10 additions and 7 deletions
--- a/src/CountriesMerger.py
+++ b/src/CountriesMerger.py
@@ -4,7 +4,10 @@ import pandas as pd
 class CountriesMerger:

    @staticmethod
-    def mergeCountriesSerieses(countriesSeriesA, countriesSeriesB):
+    def mergeSrcIntoDst(src: pd.Series, dst: pd.Series):
+        def merge(series):
+            series = series.dropna()
+            return sorted(series[0].union(series[1:]))
        return (pd
-                .merge(countriesSeriesA, countriesSeriesB, how='left', left_index=True, right_index=True)
-                .apply(lambda countries: sorted(set(countries.dropna())), axis='columns'))
+                .merge(dst, src, how='left', left_index=True, right_index=True)
+                .apply(merge, axis='columns'))