檢查變量名稱中是否存在 substring - Python

Question

我有一個 function 將給定變量作為參數。 這個變量是一個 numpy 數組。 偽代碼示例：

def foo(var):
   if 'test' in the name of var:
      do something

return

我嘗試了不同的選擇，比如

if 'test' in var: 

或者

if var == 'test_complete_name_of_var'

但這些只是檢查數組的值，而不是變量名。 我希望可能有某種技巧，例如：

if 'test' in var.name()

關於如何解決這個問題的任何想法？

Answer 1

通常，如果您想做類似的事情，您可以創建一個字典並將變量存儲在一個名稱下。 這樣你也可以檢查它。

例子：

dict_variables = {}
dict_variables["var_1"] = ["a", "b", "test"]

def foo(var_name):
    var_value = dict_variables.get(var_name)
    if 'var_' in var_name:
       print(var_value)
       
foo("var_1")
#out: ['a', 'b', 'test']

例如對於路徑和 dataframe 變量很有用。

Answer 2

一、安裝varname庫

pip install python-varname

然后將其用作以下示例：

from varname import nameof

def foo(var):
    if 'test' in nameof(var):
      ###do something