Python 3.x 如何使用单个参数从python函数返回变量

我有以下功能：

def test(crew):
  crew1 = crew_data['CrewEquipType1']
  crew2 = crew_data['CrewEquipType2']
  crew3 = crew_data['CrewEquipType3']
  return
test('crew1')

我希望能够使用这3个变量中的任何一个作为参数，并相应地返回输出，以便稍后在代码中用作引用。仅供参考，上述每个变量都是数据帧中的一个系列。 我可以在没有参数的情况下创建函数，但由于这个原因，我不能完全理解如何有效地使用上述参数，相反，我发现自己编写了单个函数，而不是编写单个函数并添加参数

---

如果有人能提供上述问题的解决方案，我们将不胜感激。

假设：您的问题似乎是希望根据对函数的输入返回相应的变量

crew1

、

crew2

或

crew3

基于我对您的问题的理解，提供了一些测试用例

• test（'crew1'）

  应返回

  crew\u数据['CrewEquipType1']

• test（'crew2'）

  应返回

  crew\u数据['CrewEquipType2']

• test（'crew3'）

  应返回

  crew\u数据['CrewEquipType3']

要实现这一点，您可以实现如下功能

def测试（乘员）： 如果乘员=='crew1'： 返回机组数据['CrewEquipType1'] elif crew=='crew2'： 返回机组数据['CrewEquipType2'] elif crew=='crew3'： 返回机组数据['CrewEquipType3'] ... ... # 添加任意数量的案例 ... 其他： #您可以在此处处理'crew'参数的不正确值 希望这有帮助！

---

如果不是，请发表评论

绝对正确！