29 lines
689 B
Python
29 lines
689 B
Python
import pandas as pd
|
|
import seaborn as sns
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
path = "/rapids/data/processed/features/all_participants/all_sensor_features.csv"
|
|
df = pd.read_csv(path)
|
|
|
|
# Bluetooth
|
|
doryab_cols_bt = [col for col in df.columns if "bluetooth_doryab" in col]
|
|
df_bt = df[doryab_cols_bt]
|
|
|
|
print(len(doryab_cols_bt))
|
|
print(df_bt)
|
|
|
|
sns.heatmap(df_bt, xticklabels=1)
|
|
plt.savefig(f'bluetooth_doryab_values', bbox_inches='tight')
|
|
|
|
# Location
|
|
doryab_cols_loc = [col for col in df.columns if "locations_doryab" in col]
|
|
df_loc = df[doryab_cols_loc]
|
|
|
|
print(len(doryab_cols_loc))
|
|
print(df_loc)
|
|
|
|
sns.heatmap(df_loc, xticklabels=1)
|
|
plt.savefig(f'locations_doryab_values', bbox_inches='tight')
|
|
|