23 lines
741 B
Python
23 lines
741 B
Python
import pandas as pd
|
|
import numpy as np
|
|
import seaborn as sns
|
|
import matplotlib.pyplot as plt
|
|
import os, sys
|
|
|
|
participant = "p032"
|
|
|
|
folder = f"/rapids/data/processed/features/{participant}/"
|
|
for filename in os.listdir(folder):
|
|
if filename.startswith("phone_"):
|
|
df = pd.read_csv(f"{folder}{filename}")
|
|
plt.figure()
|
|
sns.heatmap(df[[col for col in df if col.startswith('phone_')]], cbar=True)
|
|
plt.savefig(f'{participant}_{filename}.png', bbox_inches='tight')
|
|
plt.close()
|
|
|
|
plt.figure()
|
|
sns.heatmap(df[[col for col in df if col.startswith('phone_')]].isna(), cbar=True)
|
|
plt.savefig(f'is_na_{participant}_{filename}.png', bbox_inches='tight')
|
|
plt.close()
|
|
|
|
|