Mercurial > hg > plosone_underreview

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

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge
author Maria Panteli
date Sun, 15 Oct 2017 13:36:59 +0100
parents 7a8d043d2d45
children
line wrap: on
line source
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)