funcion_separar.py

def separar(df,nvel,ndir):
  print(df.columns)
  limite=df[[nvel]].max()/3
  print(limite.values)
  
  limite = limite.values[0]
  filtro = df[nvel] <= limite
  df1 = df.loc[filtro]
  filtro = df[ndir] <= 180
  df1 = df1.loc[filtro]

  filtro = (df[nvel] > limite) & (df[nvel] < limite*2)
  df2 = df.loc[filtro]
  filtro = df[ndir] <= 180
  df2 = df2.loc[filtro]

  filtro = df[nvel] > limite*2
  df3 = df.loc[filtro]
  filtro = df[ndir] <= 180
  df3 = df3.loc[filtro]

  filtro = df[nvel] <= limite
  df4 = df.loc[filtro]
  filtro = df[ndir] > 180
  df4 = df4.loc[filtro]

  filtro = (df[nvel] > limite) & (df[nvel] < limite*2)
  df5 = df.loc[filtro]
  filtro = df[ndir] > 180
  df5 = df5.loc[filtro]

  filtro = df[nvel] > limite*2
  df6 = df.loc[filtro]
  filtro = df[ndir] > 180
  df6 = df6.loc[filtro]


  return df1,df2,df3,df4,df5,df6
df_obj =df_corr[['V_O','D_O']]

df_reg = df_corr[['V_R','D_R']]
print(df_corr)
do1,do2,do3,do4,do5,do6 = separar(df_corr,'V_O','D_O')
dr1,dr2,dr3,dr4,dr5,dr6 = separar(df_reg,'V_R','D_R')

print(do1)
print(dr1)