opxrd/analyis: Fixed label_fractions table

aimat-lab · Dec 23, 2024 · 0667596 · 0667596
1 parent b021334
commit 0667596
Show file tree

Hide file tree

Showing 2 changed files with 295 additions and 90 deletions.
diff --git a/opxrd/analysis/tables.py b/opxrd/analysis/tables.py
@@ -25,16 +25,14 @@ def show_label_fractions(self):
         print_text(f'---> Overview of label fractions per contribution')
         table_data = []
         for d in self.databases:
-            table_data.append(len(d.patterns))
-
             label_counts = {l: 0 for l in LabelType}
             patterns = d.patterns
             for l in LabelType:
                 for p in patterns:
                     if p.has_label(label_type=l):
                         label_counts[l] += 1
-            db_percentages = [label_counts[l] / len(patterns) for l in LabelType]
-            table_data.append(db_percentages)
+            row = [len(d.patterns)] + [label_counts[l] / len(patterns) for l in LabelType]
+            table_data.append(row)
 
         col_headers = ['No. patterns'] +  [label.name for label in LabelType]
         row_headers = [db.name for db in self.databases]

diff --git a/opxrd/plots.ipynb b/opxrd/plots.ipynb