continuing

src/CountriesColumnAdder.py

@@ -0,0 +1,8 @@
+class CountriesColumnAdder:
+
+    def addCountriesColumn(self, countriesByBatchcodeTable, columnName):
+        countriesByBatchcodeTable[columnName] = countriesByBatchcodeTable.apply(self._getCountriesHavingEvents, axis='columns')
+        return countriesByBatchcodeTable
+
+    def _getCountriesHavingEvents(self, eventCountByCountry):
+        return set(eventCountByCountry[eventCountByCountry > 0].index)