Добавить столько элементов, чтобы положительных и отрицательных стало бы поровну

import random
 
N = int(input('Введите количество элементов массива: '))
arr = [int(input('Введите элемент массива: ')) for i in range(N)]
 
plus = 0
minus = 0
 
for i in range(N):
    if arr[i] > 0:
        plus += 1
    elif arr[i] < 0:
        minus += 1
 
if plus > minus:
    for i in range(minus, plus):
        arr.append(random.randint(-100, -1))
elif plus < minus:
    for i in range(plus, minus):
        arr.append(random.randit(1, 100))
        
print('Количество положительных элементов массива:', plus)
print('Количество отрицательных элементов массива:', minus)
print('Конечный массив:', arr)

N = int(input('Введите количество элементов массива: '))
arr = [int(input('Введите элемент массива: ')) for i in range(N)]
 
plus = 0
minus = 0
 
for i in range(N):
    if arr[i] > 0:
        plus += 1
    elif arr[i] < 0:
        minus += 1
 
if plus > minus:
    for i in range(minus, plus):
        arr.append(int(input('Введите отрицательное число: ')))
elif plus < minus:
    for i in range(plus, minus):
        arr.append(int(input('Введите положительное число: ')))
        
print('Количество положительных элементов массива:', plus)
print('Количество отрицательных элементов массива:', minus)
print('Конечный массив:', arr)

from random import randint
arr = [randint(-100, 100) for i in range(20)] # для примера а так по условию массив уже дан
#################################################################
 
def add_condition():
     len_arr = len(arr)
     positive = len(list(filter(lambda x : x > 0, arr)))
     negative = len_arr - positive
     factor = 1
     if positive > negative:
          factor = -1
     for _ in range( abs(positive - negative)):
          arr.append( randint(0, 100)*factor)
           
 
print(arr)
add_condition()
print(arr)