Работа с таблицами Exel (в мини проекте) v2.0

from openpyxl import load_workbook
from openpyxl.utils import get_column_letter
'''
Считывание таблицы
'''
file_obj = '/Users/markgurianov/Documents/example.xlsx'
workbook = load_workbook(filename='/Users/markgurianov/Documents/example.xlsx')
sheet = workbook.worksheets[0]
sheet_of_project = []
for row1 in range(1,sheet.max_row+1):
    stroka = []
    for column1 in range(1, sheet.max_column + 1):
        if sheet.cell(row=row1, column=column1).value is not None:
            stroka.append(int(sheet.cell(row=row1, column=column1).value))
        else:
            stroka.append(' ')
    sheet_of_project.append(stroka)
'''
Узнаём номер столбца относительно которого произойдёт сортировка
'''
base_column = int(input('Введите индекс столбца для сортировки: '))
'''
Сортировка
'''
sheet_of_project1 = sorted(sheet_of_project, key=lambda sheet_sorted: sheet_sorted[base_column])
'''
Вывод отсортированных данных в таблицу
'''
for row1 in range(1,sheet.max_row+1):
    for column1 in range(1, sheet.max_column + 1):
        coordinate = get_column_letter(column1) + str(row1)
        sheet[coordinate] = sheet_of_project1[row1 - 1][column1 - 1]
workbook.save(file_obj)

from openpyxl import load_workbook
from openpyxl.utils import get_column_letter, column_index_from_string
'''
Считывание таблицы
'''
file_obj = input('Введите путь к файлу: ') # /Users/markgurianov/Downloads/TMRCA.xlsx
workbook = load_workbook(filename=file_obj)
sheet = workbook.worksheets[0]
'''
Проверка правильности считывания
'''
row = int(input('Введите количество строк в таблице: '))
column = int(input('Введите количество столбцов в таблице: '))
if row == sheet.max_row and column == sheet.max_column:
    print('Чтение файла завершилось верно')
else:
    print('Ошибка чтения файла!!!')
'''
Получение дополнительных данных
'''
booling = 'н'
while booling == 'н':
    base_index = int(input('Введите индекс гаплотипа: '))
    print('Проверьте данные:')
    print('| ', end='')
    for column1 in range(3, sheet.max_column + 1):
        if sheet.cell(row=base_index, column=column1).value is not None:
            print('{: ^ 4} | '.format(sheet.cell(row=base_index, column=column1).value), end='')
        else:
            continue
    print()
    booling = input("Правильно ? (д/н): ")
    if booling == 'д':
        break
age = float(input('Введите средний возраст покаления: '))
'''
const
'''
mu = 0.0026
'''
Подсчитаем TMRCA в гаплоидных наборах относительно базового 
'''
new_column = sheet.max_column + 1
title_coordinate = get_column_letter(new_column) + str(1)
 
sheet[title_coordinate] = 'TMRCA'
for row1 in range(2, sheet.max_row + 1):
    diff_square = 0
    count_locus = 0
    if row1 != base_index:
        for column1 in range(3, sheet.max_column + 1):
            a = sheet.cell(row=base_index, column=column1).value
            b = sheet.cell(row=row1, column=column1).value
            if a is not None and b is not None:
                calculation = ((a - int(b)) ** 2) / 2
                diff_square += calculation
                count_locus += 1
            else:
                continue
    else:
        continue
    average_value = diff_square / count_locus
    TMRCA = age * (average_value / mu)
    coordinate = get_column_letter(new_column) + str(row1)
    sheet[coordinate] = TMRCA
 
'''
Создаём список с данными
'''
sheet_of_project = []
for row1 in range(1,sheet.max_row+1):
    stroka = []
    for column1 in range(1, sheet.max_column + 1):
        if sheet.cell(row=row1, column=column1).value is not None:
            stroka.append(int(sheet.cell(row=row1, column=column1).value))
        else:
            stroka.append(' ')
    sheet_of_project.append(stroka)
'''
Столбец относительно которого сортируем таблицу
'''
base_column = sheet.max_column + 1
'''
Сортировка
'''
sheet_of_project1 = sorted(sheet_of_project, key=lambda sheet_sorted: sheet_sorted[base_column])
'''
Вывод отсортированных данных в таблицу
'''
for row1 in range(1,sheet.max_row+1):
    for column1 in range(1, sheet.max_column + 1):
        coordinate = get_column_letter(column1) + str(row1)
        sheet[coordinate] = sheet_of_project1[row1 - 1][column1 - 1]
 
workbook.save(file_obj)