iSLR-demo / utils.py
PBJ's picture
Upload 5 files
b641e9a
raw
history blame contribute delete
337 Bytes
import numpy as np
import pandas as pd
def load_relevant_data_subset(df, ROWS_PER_FRAME):
data_columns = ['x', 'y', 'z']
data = df
n_frames = int(len(data) / ROWS_PER_FRAME)
data = data.values.reshape(n_frames, ROWS_PER_FRAME, len(data_columns))
return data.astype(np.float32)
def display_dataframe():
return df