Я думаю, что вы перезагружаете раздел после развертывания / сворачивания разделов 'tableView'.
Вы можете прокрутить tableView к прямоугольнику раздела, который вы расширяете.
Например:
let sectionRect = tableView.rect(forSection: section)
tableView.scrollRectToVisible(sectionRect, animated: false)
Надеюсь, это поможет!