refactoring
This commit is contained in:
7
src/SevereColumnAdder.py
Normal file
7
src/SevereColumnAdder.py
Normal file
@@ -0,0 +1,7 @@
|
||||
class SevereColumnAdder:
|
||||
|
||||
@staticmethod
|
||||
def addSevereColumn(dataFrame):
|
||||
dataFrame['SEVERE'] = (dataFrame['DIED'] + dataFrame['L_THREAT'] + dataFrame['DISABLE']) > 0
|
||||
dataFrame['SEVERE'].replace({True: 1, False: 0}, inplace = True)
|
||||
return dataFrame
|
||||
Reference in New Issue
Block a user