Re-wrote partitioned_artist_arrays() to use list comprehension instead of for-loop