annotate scripts/add_collection.py @ 105:edd82eb89b4b branch-tests tip

Merge

import numpy as np
import pandas as pd
import pickle 
import os


METADATA_FILE = '../../music-outliers/data/metadata.csv'
#METADATA_FILE = '../data/metadata.csv'

df = pd.read_csv(METADATA_FILE)

collection = []
for audio in df['Audio']:
    if 'SampleAudio' in audio:
        collection.append('Smithsonian Folkways')
    else:
        collection.append('British Library')
collection = np.array(collection)
df['Collection'] = collection

df.to_csv('../../music-outliers/data/metadata_2.csv', index=False)