Spaces:
Sleeping
Sleeping
File size: 1,385 Bytes
052f08d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import pandas as pd
import streamlit as st
def csv_uploader(label:str='', type:str='csv'):
uploaded_file = st.file_uploader(label=label, type=type)
loading_data = None
if uploaded_file is not None:
if type == 'csv':
loading_data = pd.read_csv(uploaded_file, na_values=-9999)
return loading_data
@st.cache_data
def tweak_data(data, reindex:bool=True, resample:bool=False, max_number:int=5000):
if data is not None:
data = data.rename(str.upper, axis='columns')
if len(data) > max_number and resample:
data = data.sample(max_number, random_state=42)
if set(["DEPTH", 'WELL']).issubset(data.columns.unique()):
data.assign(
DEPTH=data.DEPTH.astype('float64'),
WELL=data.WELL.astype("string"),
)
if "FRACTURE_ZONE" in data.columns.unique():
data = data.rename(columns={'FRACTURE_ZONE':'FRACTUREZONE'})
if "FRACTURE INTENSITY" in data.columns.unique():
data = data.rename(columns={'FRACTURE INTENSITY':'FRACTURE_INTENSITY'})
# if "FRACTUREZONE" in data.columns.unique():
# data = data.reset_index(drop=True).drop(["FRACTUREZONE"], axis=1)
if reindex:
data = data.reindex(sorted(data.columns), axis=1) #Reindex and sort columns name
return data
|