从 Python 中的字符串引用变量

Question

我的脚本中有许多全局变量（例如 group_1、group_2、group_3），我试图在函数中引用它们。

def plot(group_number):
   return(f"group_{group_number}")

在我运行plot(2)时，我希望在这种情况下函数return全局变量group_2。有没有办法把这个字符串“group_2”变成变量group_2?

Answer 1

全局变量可通过 globals() 返回的 dict 获得。所以，

>>> group_1 = "one"
>>> group_2 = "two"
>>> group_3 = "three"
>>> 
>>> def plot(group_number):
...     return globals()[f"group_{group_number}"]
... 
>>> plot(2)
'two'

从 Python 中的字符串引用变量

Reference a variable from a string in Python

python

function