9 lines
405 B
Python
9 lines
405 B
Python
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)
|