You could use ```values = values[0:1]``` to keep it as an numpy array.