Python: параметр функции между цитатами для столбцов

Question

В основном мне нужно следующее:

data['sales']*10
data['height']*10

Основная проблема, с которой я сталкиваюсь, заключается в том, как создать функцию, в которой я могу писать имена переменных без добавления кавычек внутри функции.Это возможно?Например, наподобие написания специального символа внутри «», указывающего, что слово является аргументом.

def function(var1):
   p=data['var1']*10  #The error is here; I tried p=data["'"+var1+"'"]*10
                    #Is there a way to indicate var1 is not a string,
                    #like p=data['&var1']*10
return p

function(sales)
function(height)

Я знаю, что вопрос очень простой, но мне нужно знать, возможно ли это.Если нет, я просто создам все функции и добавлю цитаты для каждого аргумента.Спасибо.

Answer 1

Вам просто нужно передать строки (имя поля / столбца) вашей функции в качестве аргумента, который будет сохранен в переменной var1.Тогда вам не нужно помещать кавычки вокруг var1 внутри вашей функции.Например, сделайте следующее

def function(var1):
   p=data[var1]*10  #The error is here; I tried p=data["'"+var1+"'"]*10
                    #Is there a way to indicate var1 is not a string,
                    #like p=data['&var1']*10
   return p

function('sales')
function('height')