Как перебирать столбцы Excel

Question

(Это может быть простой вопрос, но я довольно плохо знаком с python). У меня есть функция с несколькими аргументами (18 аргументов). У меня также есть файл Excel, который содержит входные данные, которые я хочу передать в функцию. Я предполагаю, что первым шагом является преобразование файла Excel в фрейм данных. Однако моя проблема в том, что я не знаю, как передать значения столбцов в функцию.

Моя функция выглядит следующим образом (включены 4 из 18 аргументов только для примера):

def calculation(calculation_date, redemption, spot_price, conversion_ratio):
    ... #calculations here
    return bond_data

Мои данные структурированы так:

Как передать значения из каждого столбца в функцию? Я структурировал свой файл данных так, чтобы первый столбец (столбец A) соответствовал первому аргументу, а второй столбец (столбец B) соответствовал второму аргументу et c?

Answer 1

Вы можете попробовать это

После чтения файла Excel, если у вас есть df в качестве объекта фрейма данных, тогда

column_names=list(df.columns)

column_names будет содержать список столбцов, которые у вас есть в вашем файле Excel. Теперь вы можете перебирать его. пример

for column_name in column_names:
      #your code

Примечание: вы можете напрямую передать объект списка column_names в функцию.

Пример:

#function defination
def column_names(list_of_columns)

#calling function
calculation(column_names) 

дайте мне знать если это работает для вас или нет.

Спасибо

Answer 2

проверьте это: Руководство по таблицам Excel в Python С openpyxl

вы можете сделать что-то вроде этого:

from openpyxl import load_workbook

workbook = load_workbook(filename="sample.xlsx")
sheet = workbook.active
for row in sheet.iter_rows(min_row=2,
                       min_col=1,
                       max_col=4,
                       values_only=True):
    row_dic = {
               "parent": row[1],
               "title": row[2],
               "category": row[3]}
    excel_list.append(row_dic)

, и вы можете использовать список для вызова функции расчета.