Extract categories from CMU

parse.py

import pandas as pd

table = pd.read_csv('cmu-mocap-index-spreadsheet.csv', sep='\t')
table = table.dropna(axis=0)
table.columns = ['Motion', 'Description', 'Subject']

categories = [
    'walk', 'run|jog', 'jump', 'dance', 'sit', 'climb', 'balance', 'hop', 'stretch', 'flip',
    'basketball', 'soccer', 'boxing',
    'drink', 'wash windows', 'bend', 'cartwheel',
    'wave', 'limp',
]

temp_table = table.copy()

by_category = {}
for category in categories:
    does_contain = temp_table['Description'].str.contains(category, case=True)
    by_category[category] = temp_table[does_contain]
    temp_table = temp_table[~does_contain]

result = {k: list(table['Motion']) for k, table in by_category.items()}

1. Install pandas
2. Get CMU Excel file and convert it into csv (using tab delimiter: \t)
3. Use this ocde