Python中用input函数的情况下如何才能调出对应编号的数据？

img


请问各位，最后那部分要如何写才能在输入数字的情况下显示出对应的身高？

print("第{}个人的身高为{}".format(str, person_height[int(str)]))

person = ["0:李","1:张","2:王","3:吴","4:马"]
str = input()
person_height = [182,143,95,173,133]
print(f'{[a[2:3] for a in person if a.startswith(str)][0]}的人的身高为{person_height[int(str)]}cm')

img