Практика программирования с помощью функции

У меня есть функция, которая возвращает целое число , однако я хотел бы расширить ее, чтобы добавить к ней новый параметр. Однако с этим параметром функция должна возвращать массив .

• Является ли плохой практикой иметь функцию, которая возвращает либо массив, либо целое число на основе параметра?
• Если так , как я могу это решить?

Я думаю, что было бы плохой практикой также просто скопировать и вставить всю функцию для 4-5 дополнительных строк.