Maria@103: import numpy as np
Maria@103: import pandas as pd
Maria@103: import pickle 
Maria@103: import os
Maria@103: 
Maria@103: 
Maria@103: METADATA_FILE = '../../music-outliers/data/metadata.csv'
Maria@103: #METADATA_FILE = '../data/metadata.csv'
Maria@103: 
Maria@103: df = pd.read_csv(METADATA_FILE)
Maria@103: 
Maria@103: collection = []
Maria@103: for audio in df['Audio']:
Maria@103:     if 'SampleAudio' in audio:
Maria@103:         collection.append('Smithsonian Folkways')
Maria@103:     else:
Maria@103:         collection.append('British Library')
Maria@103: collection = np.array(collection)
Maria@103: df['Collection'] = collection
Maria@103: 
Maria@103: df.to_csv('../../music-outliers/data/metadata_2.csv', index=False)