init depot renew
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
import pandas as pd
|
||||
import numpy as np
|
||||
import os
|
||||
|
||||
path = '/home/colas/Documents/5_BDD/3_INPN/BDD_ZNIEFF'
|
||||
reg = '084_Auvergne-Rhone-Alpes'
|
||||
ssreg = '082_Rhone-Alpes'
|
||||
|
||||
DIR = os.path.join(path,reg,ssreg)
|
||||
|
||||
|
||||
def znieff(path,code,column='NM_REGZN'):
|
||||
df = pd.read_csv(os.path.join(path,'znieff.csv'),sep=';')
|
||||
df = df.astype(str)
|
||||
|
||||
if isinstance(code,str):
|
||||
return df[df[column]==code]
|
||||
elif isinstance(code,(pd.Series,list,pd.Index,np.ndarray)):
|
||||
code = list(code)
|
||||
return df[df[column].isin(code)]
|
||||
else:
|
||||
return df
|
||||
|
||||
|
||||
def source_znieff(path,code,column='NM_SFFZN'):
|
||||
df = pd.read_csv(os.path.join(path,'source_znieff.csv'),sep=';')
|
||||
df = df.astype(str)
|
||||
|
||||
if isinstance(code,str):
|
||||
return df[df[column]==code]
|
||||
elif isinstance(code,(pd.Series,list,pd.Index,np.ndarray)):
|
||||
code = list(code)
|
||||
return df[df[column].isin(code)]
|
||||
else:
|
||||
return df
|
||||
|
||||
|
||||
def source(path,code,column='ID_SOURCE'):
|
||||
df = pd.read_csv(os.path.join(path,'source.csv'),sep=';')
|
||||
df = df.astype(str)
|
||||
|
||||
if isinstance(code,str):
|
||||
return df[df[column]==code]
|
||||
elif isinstance(code,(pd.Series,list,pd.Index,np.ndarray)):
|
||||
code = list(code)
|
||||
return df[df[column].isin(code)]
|
||||
else:
|
||||
return df
|
||||
|
||||
|
||||
def _rel_znieff_source(path,code,column='NM_REGZN'):
|
||||
znief = znieff(path,code,column)
|
||||
sznief = source_znieff(path,znief.NM_SFFZN.unique())
|
||||
|
||||
return source(path,sznief.ID_SOURCE.unique())
|
||||
|
||||
|
||||
def get_source(path,code):
|
||||
return _rel_znieff_source(path,code)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
get_source(DIR,'38000064')
|
||||
Reference in New Issue
Block a user