import panda as pd

data = pd.read_csv("fer2013.csv")

#-------------------------------------------------------------
#Extract emotions 0. Anger, 3. Happy, 6. Neutral original file
#And store in new csv file for first iteration
#-------------------------------------------------------------

test = data.loc[data["emotion"].isin(['0', '3', '6'])
dx = test.to_numpy()

test.to_csv('DataSplit.csv', sep='\t')
print(test)