diff options
| author | Aaron LI <aaronly.me@outlook.com> | 2016-05-01 17:58:24 +0800 |
|---|---|---|
| committer | Aaron LI <aaronly.me@outlook.com> | 2016-05-01 17:58:24 +0800 |
| commit | 26251d7ca84267906e8810b186627427e790d18f (patch) | |
| tree | 029d07cd2aa2d4f6d89ac8fde0e0ed17bac3ac8a | |
| parent | e7106fdc4354bed002ec5bcfd39b8a03e96ae04e (diff) | |
| download | cexcess-26251d7ca84267906e8810b186627427e790d18f.tar.bz2 | |
calc_radial_psd.py: adjust axis limits for plot
| -rwxr-xr-x | calc_radial_psd.py | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/calc_radial_psd.py b/calc_radial_psd.py index 23bd819..4c1d85b 100755 --- a/calc_radial_psd.py +++ b/calc_radial_psd.py @@ -13,9 +13,11 @@ # # Aaron LI <aaronly.me@gmail.com> # Created: 2015-04-22 -# Updated: 2016-04-28 +# Updated: 2016-05-01 # # Changelog: +# 2016-05-01: +# * Adjust plot axis limits # 2016-04-28: # * Fix wrong meshgrid with respect to the shift zero-frequency component # * Use "numpy.fft" instead of "scipy.fftpack" @@ -41,8 +43,8 @@ Compute the radially averaged power spectral density (i.e., power spectrum). """ -__version__ = "0.5.0" -__date__ = "2016-04-28" +__version__ = "0.5.1" +__date__ = "2016-05-01" import sys @@ -229,10 +231,11 @@ class PSD: if ax is None: fig, ax = plt.subplots(1, 1) # - xmin = self.freqs[1] / 1.2 # ignore the first 0 - xmax = self.freqs[-1] - ymin = np.nanmin(self.psd1d) / 10.0 - ymax = np.nanmax(self.psd1d + self.psd1d_err) + mask = (self.freqs > 0.0) + xmin = np.min(self.freqs[mask]) / 1.2 + xmax = np.max(self.freqs[mask]) + ymin = np.min(self.psd1d) / 3.0 + ymax = np.max(self.psd1d[mask] + self.psd1d_err[mask]) * 1.2 # eb = ax.errorbar(self.freqs, self.psd1d, yerr=self.psd1d_err, fmt="none") @@ -447,4 +450,3 @@ def main(): if __name__ == "__main__": main() - |