AeroViz 0.1.3__py3-none-any.whl → 0.1.3b0__py3-none-any.whl

AeroViz/dataProcess/Chemistry/_partition.py

@@ -1,29 +1,30 @@
-from pandas import date_range, concat, DataFrame, to_numeric
+from pandas import concat, DataFrame
+
 from ._calculate import _ug2umol
 
 
 def _basic(df_che, nam_lst):
-	# parameter
-	df_all = concat(df_che, axis=1)
-	index = df_all.index.copy()
-	df_all.columns = nam_lst
+    # parameter
+    df_all = concat(df_che, axis=1)
+    index = df_all.index.copy()
+    df_all.columns = nam_lst
 
-	df_umol = _ug2umol(df_all)
+    df_umol = _ug2umol(df_all)
 
-	# calculate
-	df_out = DataFrame(index=df_umol.index)
+    # calculate
+    df_out = DataFrame(index=df_umol.index)
 
-	# df_out['NTR'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_all['NH3'] / 22.4)
-	df_out['NTR+'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_umol['NH3'])
+    # df_out['NTR'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_all['NH3'] / 22.4)
+    df_out['NTR+'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_umol['NH3'])
 
-	df_out['NOR'] = df_umol['NO3-'] / (df_umol['NO3-'] + df_umol['NO2'])
-	df_out['NOR_2'] = (df_umol['NO3-'] + df_umol['HNO3']) / (df_umol['NO3-'] + df_umol['NO2'] + df_umol['HNO3'])
+    df_out['NOR'] = df_umol['NO3-'] / (df_umol['NO3-'] + df_umol['NO2'])
+    df_out['NOR_2'] = (df_umol['NO3-'] + df_umol['HNO3']) / (df_umol['NO3-'] + df_umol['NO2'] + df_umol['HNO3'])
 
-	df_out['SOR'] = df_umol['SO42-'] / (df_umol['SO42-'] + df_umol['SO2'])
+    df_out['SOR'] = df_umol['SO42-'] / (df_umol['SO42-'] + df_umol['SO2'])
 
-	df_out['epls_NO3-'] = df_umol['NO3-'] / (df_umol['NO3-'] + df_umol['HNO3'])
-	df_out['epls_NH4+'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_umol['NH3'])
-	df_out['epls_SO42-'] = df_out['SOR']
-	df_out['epls_Cl-'] = df_umol['Cl-'] / (df_umol['Cl-'] + df_umol['HCl'])
+    df_out['epls_NO3-'] = df_umol['NO3-'] / (df_umol['NO3-'] + df_umol['HNO3'])
+    df_out['epls_NH4+'] = df_umol['NH4+'] / (df_umol['NH4+'] + df_umol['NH3'])
+    df_out['epls_SO42-'] = df_out['SOR']
+    df_out['epls_Cl-'] = df_umol['Cl-'] / (df_umol['Cl-'] + df_umol['HCl'])
 
-	return df_out
+    return df_out

AeroViz/dataProcess/Chemistry/_teom.py

@@ -1,16 +1,13 @@
-import numpy as np
-
-
 def _basic(_teom, _check):
-	_teom['Volatile_Fraction'] = (_teom['PM_Total'] - _teom['PM_NV']) / _teom['PM_Total']
+    _teom['Volatile_Fraction'] = (_teom['PM_Total'] - _teom['PM_NV']) / _teom['PM_Total']
 
-	_teom.loc[(_teom['Volatile_Fraction'] < 0) | (_teom['Volatile_Fraction'] > 1)] = n.nan
+    _teom.loc[(_teom['Volatile_Fraction'] < 0) | (_teom['Volatile_Fraction'] > 1)] = n.nan
 
-	if _check is not None:
-		_ratio = _teom['PM_NV'] / _check
-		_teom['PM_Check'] = _check
+    if _check is not None:
+        _ratio = _teom['PM_NV'] / _check
+        _teom['PM_Check'] = _check
 
-		_teom.loc[_teom.dropna().index, 'status'] = 'Warning'
-		_teom.loc[(_ratio > 0) & (_ratio < 1), 'status'] = 'Normal'
+        _teom.loc[_teom.dropna().index, 'status'] = 'Warning'
+        _teom.loc[(_ratio > 0) & (_ratio < 1), 'status'] = 'Normal'
 
-	return _teom
+    return _teom

AeroViz/dataProcess/Optical/_IMPROVE.py

@@ -1,61 +1,62 @@
 import pickle as pkl
 from pathlib import Path
+
 import numpy as np
+from pandas import DataFrame
 
-from pandas import date_range, concat, DataFrame, to_numeric
 from AeroViz.dataProcess.core import _union_index
 
 
 def _revised(_df_mass, _df_RH):
-	_df_mass, _df_RH = _union_index(_df_mass, _df_RH)
+    _df_mass, _df_RH = _union_index(_df_mass, _df_RH)
 
-	## fRH
-	with (Path(__file__).parent / 'fRH.pkl').open('rb') as f:
-		_fRH = pkl.load(f)
-		_fRH.loc[np.nan] = np.nan
+    ## fRH
+    with (Path(__file__).parent / 'fRH.pkl').open('rb') as f:
+        _fRH = pkl.load(f)
+        _fRH.loc[np.nan] = np.nan
 
-	def fRH(_RH):
-		if _RH is not None:
-			_RH = _RH.mask(_RH > 95, 95).round(0)
-			return _fRH.loc[_RH].values.T
+    def fRH(_RH):
+        if _RH is not None:
+            _RH = _RH.mask(_RH > 95, 95).round(0)
+            return _fRH.loc[_RH].values.T
 
-		return 1, 1, 1, 1
+        return 1, 1, 1, 1
 
-	## different mode
-	## mass < 20 :
-	## 				large = mass**2/20
-	## 				small = mass-large
-	## mass >= 20 :
-	## 				large = mass
-	## 				small = 0
-	_df_mode = _df_mass[['AS', 'AN', 'OM']].copy()
+    ## different mode
+    ## mass < 20 :
+    ## 				large = mass**2/20
+    ## 				small = mass-large
+    ## mass >= 20 :
+    ## 				large = mass
+    ## 				small = 0
+    _df_mode = _df_mass[['AS', 'AN', 'OM']].copy()
 
-	_df_mass[['L_AS', 'L_AN', 'L_OM']] = _df_mode.mask(_df_mode < 20, _df_mode ** 2 / 20)
-	_df_mass[['S_AS', 'S_AN', 'S_OM']] = _df_mode.values - _df_mass[['L_AS', 'L_AN', 'L_OM']]
+    _df_mass[['L_AS', 'L_AN', 'L_OM']] = _df_mode.mask(_df_mode < 20, _df_mode ** 2 / 20)
+    _df_mass[['S_AS', 'S_AN', 'S_OM']] = _df_mode.values - _df_mass[['L_AS', 'L_AN', 'L_OM']]
 
-	## apply IMPROVE ccoe.
-	def _ext_cal(_RH=None):
+    ## apply IMPROVE ccoe.
+    def _ext_cal(_RH=None):
 
-		_frh, _frhss, _frhs, _frhl = fRH(_RH)
-		_df = DataFrame(index=_df_mass.index)
+        _frh, _frhss, _frhs, _frhl = fRH(_RH)
+        _df = DataFrame(index=_df_mass.index)
 
-		_df['AS'] = 2.2 * _frhs * _df_mass['S_AS'] + 4.8 * _frhl * _df_mass['L_AS']
-		_df['AN'] = 2.4 * _frhs * _df_mass['S_AN'] + 5.1 * _frhl * _df_mass['L_AN']
-		_df['OM'] = 2.8 * _df_mass['S_OM'] + 6.1 * _frhl * _df_mass['L_OM']
-		_df['Soil'] = _df_mass['Soil']
-		_df['SS'] = 1.7 * _frhss * _df_mass['SS']
-		_df['EC'] = 10 * _df_mass['EC']
+        _df['AS'] = 2.2 * _frhs * _df_mass['S_AS'] + 4.8 * _frhl * _df_mass['L_AS']
+        _df['AN'] = 2.4 * _frhs * _df_mass['S_AN'] + 5.1 * _frhl * _df_mass['L_AN']
+        _df['OM'] = 2.8 * _df_mass['S_OM'] + 6.1 * _frhl * _df_mass['L_OM']
+        _df['Soil'] = _df_mass['Soil']
+        _df['SS'] = 1.7 * _frhss * _df_mass['SS']
+        _df['EC'] = 10 * _df_mass['EC']
 
-		_df['total'] = _df.sum(axis=1)
+        _df['total'] = _df.sum(axis=1)
 
-		return _df.dropna().reindex(_df_mass.index)
+        return _df.dropna().reindex(_df_mass.index)
 
-	## calculate
-	_out = {}
+    ## calculate
+    _out = {}
 
-	_out['dry'] = _ext_cal()
+    _out['dry'] = _ext_cal()
 
-	if _df_RH is not None:
-		_out['wet'] = _ext_cal(_df_RH)
+    if _df_RH is not None:
+        _out['wet'] = _ext_cal(_df_RH)
 
-	return _out
+    return _out

AeroViz/dataProcess/Optical/__init__.py

@@ -2,61 +2,61 @@ from ..core import _writter, _run_process
 
 __all__ = [
 
-	'Optical',
+    'Optical',
 
 ]
 
 
 class Optical(_writter):
 
-	## scatter
-	@_run_process('Optical - SAE', 'SAE')
-	def SAE(self, df_sca):
-		from ._scattering import _SAE
+    ## scatter
+    @_run_process('Optical - SAE', 'SAE')
+    def SAE(self, df_sca):
+        from ._scattering import _SAE
 
-		out = _SAE(df_sca)
+        out = _SAE(df_sca)
 
-		return self, out
+        return self, out
 
-	## absorption
-	@_run_process('Optical - absCoe', 'absCoe')
-	def absCoe(self, df_ae33, abs_band=[550]):
-		from ._absorption import _absCoe
+    ## absorption
+    @_run_process('Optical - absCoe', 'absCoe')
+    def absCoe(self, df_ae33, abs_band=[550]):
+        from ._absorption import _absCoe
 
-		out = _absCoe(df_ae33, abs_band)
+        out = _absCoe(df_ae33, abs_band)
 
-		return self, out
+        return self, out
 
-	@_run_process('Optical - AAE', 'AAE')
-	def AAE(self, df_abs):
-		from ._absorption import _AAE
+    @_run_process('Optical - AAE', 'AAE')
+    def AAE(self, df_abs):
+        from ._absorption import _AAE
 
-		out = _AAE(df_abs)
+        out = _AAE(df_abs)
 
-		return self, out
+        return self, out
 
-	## extinction
-	@_run_process('Optical - basic', 'opt_basic')
-	def basic(self, df_abs, df_sca, df_ec=None, df_mass=None, df_no2=None):
-		from ._extinction import _basic
+    ## extinction
+    @_run_process('Optical - basic', 'opt_basic')
+    def basic(self, df_abs, df_sca, df_ec=None, df_mass=None, df_no2=None):
+        from ._extinction import _basic
 
-		out = _basic(df_abs, df_sca, df_ec, df_mass, df_no2)
+        out = _basic(df_abs, df_sca, df_ec, df_mass, df_no2)
 
-		return self, out
+        return self, out
 
-	@_run_process('Optical - Mie', 'Mie')
-	def Mie(self, df_psd, df_m, wave_length=550):
-		from ._mie import _mie
+    @_run_process('Optical - Mie', 'Mie')
+    def Mie(self, df_psd, df_m, wave_length=550):
+        from ._mie import _mie
 
-		out = _mie(df_psd, df_m, wave_length)
+        out = _mie(df_psd, df_m, wave_length)
 
-		return self, out
+        return self, out
 
-	@_run_process('Optical - IMPROVE', 'IMPROVE')
-	def IMPROVE(self, df_mass, df_RH, method='revised'):
-		# _fc = __import__(f'_IMPROVE._{method}')
-		from ._IMPROVE import _revised
+    @_run_process('Optical - IMPROVE', 'IMPROVE')
+    def IMPROVE(self, df_mass, df_RH, method='revised'):
+        # _fc = __import__(f'_IMPROVE._{method}')
+        from ._IMPROVE import _revised
 
-		out = _revised(df_mass, df_RH)
+        out = _revised(df_mass, df_RH)
 
-		return self, out
+        return self, out

AeroViz/dataProcess/Optical/_absorption.py

@@ -1,54 +1,54 @@
 def _absCoe(df, abs_band):
-	import numpy as n
-	from scipy.optimize import curve_fit
+    import numpy as n
+    from scipy.optimize import curve_fit
 
-	band = n.array([370, 470, 520, 590, 660, 880, 950])
+    band = n.array([370, 470, 520, 590, 660, 880, 950])
 
-	df_out = {}
+    df_out = {}
 
-	def _get_slope(__df):
-		func = lambda _x, _sl, _int: _sl * _x + _int
-		popt, pcov = curve_fit(func, band, __df.values)
+    def _get_slope(__df):
+        func = lambda _x, _sl, _int: _sl * _x + _int
+        popt, pcov = curve_fit(func, band, __df.values)
 
-		return func(n.array(abs_band), *popt)
+        return func(n.array(abs_band), *popt)
 
-	MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
-	df_abs = (df.copy() * MAE).dropna().copy()
+    MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
+    df_abs = (df.copy() * MAE).dropna().copy()
 
-	df_out = df_abs.apply(_get_slope, axis=1, result_type='expand').reindex(df.index)
-	df_out.columns = [f'abs_{_band}' for _band in abs_band]
+    df_out = df_abs.apply(_get_slope, axis=1, result_type='expand').reindex(df.index)
+    df_out.columns = [f'abs_{_band}' for _band in abs_band]
 
-	df_out['eBC'] = df['BC6']
+    df_out['eBC'] = df['BC6']
 
-	return df_out
+    return df_out
 
 
 def _AAE(df):
-	import numpy as n
-	from scipy.optimize import curve_fit
+    import numpy as n
+    from scipy.optimize import curve_fit
 
-	def _AAEcalc(_df):
-		## parameter
-		MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
-		band = n.array([370, 470, 520, 590, 660, 880, 950])
-		_df *= MAE
+    def _AAEcalc(_df):
+        ## parameter
+        MAE = n.array([18.47, 14.54, 13.14, 11.58, 10.35, 7.77, 7.19]) * 1e-3
+        band = n.array([370, 470, 520, 590, 660, 880, 950])
+        _df *= MAE
 
-		## 7 pts fitting
-		## function
-		def _get_slope(__df):
-			func = lambda _x, _sl, _int: _sl * _x + _int
-			popt, pcov = curve_fit(func, n.log(band), n.log(__df))
+        ## 7 pts fitting
+        ## function
+        def _get_slope(__df):
+            func = lambda _x, _sl, _int: _sl * _x + _int
+            popt, pcov = curve_fit(func, n.log(band), n.log(__df))
 
-			return popt
+            return popt
 
-		## calculate
-		_AAE = _df.apply(_get_slope, axis=1, result_type='expand')
-		_AAE.columns = ['slope', 'intercept']
+        ## calculate
+        _AAE = _df.apply(_get_slope, axis=1, result_type='expand')
+        _AAE.columns = ['slope', 'intercept']
 
-		return _AAE
+        return _AAE
 
-	df_out = _AAEcalc(df[['BC1', 'BC2', 'BC3', 'BC4', 'BC5', 'BC6', 'BC7']].dropna())
-	df_out = df_out.mask((-df_out.slope < 0.8) | (-df_out.slope > 2.)).copy()
+    df_out = _AAEcalc(df[['BC1', 'BC2', 'BC3', 'BC4', 'BC5', 'BC6', 'BC7']].dropna())
+    df_out = df_out.mask((-df_out.slope < 0.8) | (-df_out.slope > 2.)).copy()
 
-	df_out['eBC'] = df['BC6']
-	return df_out.reindex(df.index)
+    df_out['eBC'] = df['BC6']
+    return df_out.reindex(df.index)

AeroViz/dataProcess/Optical/_extinction.py

@@ -1,36 +1,37 @@
-from AeroViz.dataProcess.core import _union_index
 from pandas import DataFrame
 
+from AeroViz.dataProcess.core import _union_index
+
 
 def _basic(df_abs, df_sca, df_ec, df_mass, df_no2):
-	df_abs, df_sca, df_ec, df_mass, df_no2 = _union_index(df_abs, df_sca, df_ec, df_mass, df_no2)
+    df_abs, df_sca, df_ec, df_mass, df_no2 = _union_index(df_abs, df_sca, df_ec, df_mass, df_no2)
 
-	df_out = DataFrame()
+    df_out = DataFrame()
 
-	## abs and sca coe
-	df_out['abs'] = df_abs.copy()
-	df_out['sca'] = df_sca.copy()
+    ## abs and sca coe
+    df_out['abs'] = df_abs.copy()
+    df_out['sca'] = df_sca.copy()
 
-	## extinction coe.
-	df_out['ext'] = df_out['abs'] + df_out['sca']
+    ## extinction coe.
+    df_out['ext'] = df_out['abs'] + df_out['sca']
 
-	## SSA
-	df_out['SSA'] = df_out['sca'] / df_out['ext']
+    ## SSA
+    df_out['SSA'] = df_out['sca'] / df_out['ext']
 
-	## MAE, MSE, MEE
-	if df_mass is not None:
-		df_out['MAE'] = df_out['abs'] / df_mass
-		df_out['MSE'] = df_out['sca'] / df_mass
-		df_out['MEE'] = df_out['MSE'] + df_out['MAE']
+    ## MAE, MSE, MEE
+    if df_mass is not None:
+        df_out['MAE'] = df_out['abs'] / df_mass
+        df_out['MSE'] = df_out['sca'] / df_mass
+        df_out['MEE'] = df_out['MSE'] + df_out['MAE']
 
-	## gas absorbtion
-	if df_no2 is not None:
-		df_out['abs_gas'] = df_no2 * .33
-		df_out['sca_gas'] = 10
-		df_out['ext_all'] = df_out['ext'] + df_out['abs_gas'] + df_out['sca_gas']
+    ## gas absorbtion
+    if df_no2 is not None:
+        df_out['abs_gas'] = df_no2 * .33
+        df_out['sca_gas'] = 10
+        df_out['ext_all'] = df_out['ext'] + df_out['abs_gas'] + df_out['sca_gas']
 
-	## other
-	if df_ec is not None:
-		df_out['eBC'] = df_ec / 1e3
+    ## other
+    if df_ec is not None:
+        df_out['eBC'] = df_ec / 1e3
 
-	return df_out
+    return df_out

AeroViz/dataProcess/Optical/_mie.py

@@ -2,15 +2,14 @@
 # from PyMieScatt import Mie_SD
 
 from ._mie_sd import Mie_SD
-from pandas import date_range, concat, DataFrame, to_numeric
 
 
 def _mie(_psd_ori, _RI_ori, _wave):
-	_ori_idx = _psd_ori.index.copy()
-	_cal_idx = _psd_ori.loc[_RI_ori.dropna().index].dropna(how='all').index
+    _ori_idx = _psd_ori.index.copy()
+    _cal_idx = _psd_ori.loc[_RI_ori.dropna().index].dropna(how='all').index
 
-	_psd, _RI = _psd_ori.loc[_cal_idx], _RI_ori.loc[_cal_idx]
+    _psd, _RI = _psd_ori.loc[_cal_idx], _RI_ori.loc[_cal_idx]
 
-	_out = Mie_SD(_RI.values, 550, _psd)
+    _out = Mie_SD(_RI.values, 550, _psd)
 
-	return _out.reindex(_ori_idx)
+    return _out.reindex(_ori_idx)