python - Which datatypes do I have to use to write this function in ...