write training weight in write_parametrized

scripts/write_parametrized.py

@@ -46,6 +46,7 @@ if __name__ == "__main__":
             params = tuple([float(i) for i in match.groups()])
             if not params in count_dict:
                 count_dict[params] = 0
+            # TODO: might be better to use sum of weights
             count_dict[params] += tree.GetEntries()
     f.Close()
 
@@ -71,9 +72,11 @@ if __name__ == "__main__":
                     param_values = numbers[rnd_idx]
                     for param_idx, param_name in enumerate(param_names):
                         df[param_name] = param_values[:,param_idx]
+                    df["training_weight"] = df["eventWeight"]*df["genWeight"]
                 else:
                     for param_name, param_value in zip(param_names, match_signal.groups()):
                         df[param_name] = float(param_value)
+                    df["training_weight"] = df["eventWeight"]
                 if match_signal is None:
                     out_tree_name = tree_name
                 else: