@@ -559,6 +559,7 @@ def __init__(
559
559
mirror_spectrum : bool = False ,
560
560
relative_intensity : bool = False ,
561
561
bin_peaks : Union [Literal ["auto" ], bool ] = "auto" ,
562
+ bin_method : Literal ['none' , 'sturges' , 'freedman-diaconis' ] = 'freedman-diaconis' ,
562
563
num_x_bins : int = 50 ,
563
564
peak_color : str | None = None ,
564
565
annotate_top_n_peaks : int | None | Literal ["all" ] = 5 ,
@@ -579,9 +580,14 @@ def __init__(
579
580
self .mirror_spectrum = mirror_spectrum
580
581
self .relative_intensity = relative_intensity
581
582
self .bin_peaks = bin_peaks
583
+ self .bin_method = bin_method
582
584
if self .bin_peaks == "auto" :
583
- # self.num_x_bins = sturges_rule(data, x)
584
- self .num_x_bins = freedman_diaconis_rule (data , x )
585
+ if self .bin_method == 'sturges' :
586
+ self .num_x_bins = sturges_rule (data , x )
587
+ elif self .bin_method == 'freedman-diaconis' :
588
+ self .num_x_bins = freedman_diaconis_rule (data , x )
589
+ elif self .bin_method == 'none' :
590
+ self .num_x_bins = num_x_bins
585
591
else :
586
592
self .num_x_bins = num_x_bins
587
593
self .peak_color = peak_color
0 commit comments