Saving a Python dictionary holding a numpy array to a file