Update perfect_arr.py

afif004 · web-flow · commit 89e36b6dc926 · 2024-08-05T04:11:12.000+06:00
diff --git a/Functions/array/perfect_arr.py b/Functions/array/perfect_arr.py
@@ -9,17 +9,19 @@ def perfect_number(n):
         return perfect
     return None
 
-N = int(input("Number of perfect numbers to generate: "))
-perfect_numbers = []
-n = 2
-pnum_count = 0
-
-while pnum_count < N:
-    perfect_num = perfect_number(n)
-    if perfect_num:
-        perfect_numbers.append(perfect_num)
-        pnum_count += 1
-    n += 1
+def perfect_num_generate(N):
+    perfect_numbers = []
+    n = 2
+    pnum_count = 0
+    while pnum_count < N:
+        perfect_num = perfect_number(n)
+        if perfect_num:
+            perfect_numbers.append(perfect_num)
+            pnum_count += 1
+        n += 1
+    return perfect_numbers
 
+N = int(input("Number of perfect numbers to generate: "))
+perfect_numbers = perfect_num_generate(N)
 print(f"Perfect number: {perfect_numbers[N-1]}")
 print("All perfect numbers generated:", perfect_numbers)
