Mercurial > hg > plosone_underreview

view scripts/add_collection.py @ 103:7a8d043d2d45 branch-tests

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
more on metadata for public code
author Maria Panteli
date Thu, 05 Oct 2017 11:47:26 +0100
parents
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)