In [1]:

    

import numpy as np
import pandas as pd

import os


    

In [5]:

    

DATA_DIRECTORY = "Q:\\p_eaglesense\\eaglesense\\data\\topviewkinect"


    

In [6]:

    

labels_csv_f = "Q:\\p_eaglesense\\eaglesense\\data\\topviewkinect\\{id}\\labels.csv"
old_labels_csv_f = "Q:\\p_eaglesense\\eaglesense\\data\\topviewkinect\\{id}\\labels - Copy.csv"


    

In [ ]:

    

for dataset_id in next(os.walk(DATA_DIRECTORY))[1]:
    if not dataset_id.isdigit():
        continue
    elif dataset_id.startswith(("1", "3", "4", "6", "7")):
        continue
    else:
        print(dataset_id, "... ", end="")

    labels_csv = labels_csv_f.format(id=dataset_id)
    labels_df = pd.read_csv(labels_csv)
    
    old_labels_csv = old_labels_csv_f.format(id=dataset_id)
    old_labels_df = pd.read_csv(old_labels_csv)
    
    common_frames = np.intersect1d(labels_df["frame_id"].values, old_labels_df["frame_id"].values)
    old_labels = old_labels_df.loc[old_labels_df["frame_id"].isin(common_frames), "activity"]
    labels_df.loc[labels_df["frame_id"].isin(common_frames), "activity"] = old_labels.astype(np.int)
#     labels_df["activity"] = labels_df["activity"].astype(np.int)
    labels_df.to_csv(labels_csv, index=False)