<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:

def split_data(data, subset, splits):
  '''Returns the data for a given protocol
  '''

  return dict([(k, data[k][splits[subset]]) for k in data])