refactoring

2023-08-23 21:28:40 +02:00
parent af19dc710f
commit 3e1c59a697
3 changed files with 27 additions and 23 deletions
--- a/src/JensenShannonDistance2BarChartDescriptionColumnAdder.py
+++ b/src/JensenShannonDistance2BarChartDescriptionColumnAdder.py
@@ -0,0 +1,19 @@
+from scipy.spatial import distance
+
+
+class JensenShannonDistance2BarChartDescriptionColumnAdder:
+
+    @staticmethod
+    def addJensenShannonDistance2BarChartDescriptionColumn(barChartDescriptionTable):
+        barChartDescriptionTable['BAR_CHART_DESCRIPTION'] = (
+            barChartDescriptionTable.apply(
+                lambda barChartDescription:
+                    {
+                        **barChartDescription['BAR_CHART_DESCRIPTION'],
+                        'Jensen-Shannon distance': distance.jensenshannon(
+                            barChartDescription['BAR_CHART_DESCRIPTION']['Adverse Reaction Reports guessed'],
+                            barChartDescription['BAR_CHART_DESCRIPTION']['Adverse Reaction Reports known'],
+                            base=2.0)
+                    },
+                axis='columns'))
+        return barChartDescriptionTable