diff --git a/earthquakes.py b/earthquakes.py index c7fb594..be20fa8 100644 --- a/earthquakes.py +++ b/earthquakes.py @@ -1,8 +1,29 @@ #! /usr/bin/env python +import pandas as pd +import json from parser import parse +def guardar_df(df: pd.DataFrame, fname: str) -> bool: + with open(fname, "w") as fp: + fname = f"{fname}.txt" + try: + fp.write(df.to_string()) + except ValueError: + return False + return True + +def guardar_json(df: pd.DataFrame, fname: str) -> bool: + with open(fname , "w") as fp: + try: + json.dump(df.to_json(), fp) + except: + return False + return True + def main(): + pass + diff --git a/parser.py b/parser.py index 8875ca7..03456fd 100644 --- a/parser.py +++ b/parser.py @@ -32,12 +32,6 @@ def into_dataframe(data) -> pd.DataFrame: return pd.DataFrame(data=aux) - -def test(d1, d2): - for col in d2.columns: - d1.at[0, col] = d2[col].tolist() - return d1 - # ------------ principal def parse(fname="dados.txt"): @@ -49,7 +43,9 @@ def parse(fname="dados.txt"): a = parse_chunk(data[c[0]:c[1]], idx) aux = pd.concat([df, a], axis=0, ignore_index=True) df = aux - + print(df) + aux = df.loc[df["ID"] == 14] + print(aux) fp.close() return df @@ -80,7 +76,6 @@ def parse_chunk(chunk_lines: list[str], iD): hDF = into_dataframe(headersRet) hDF["ID"] = iD phaseRet["ID"] = iD - return pd.concat([hDF, phaseRet])