File size: 337 Bytes
b641e9a
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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