how can I rewrite the code using return statement?

Question

https://code.sololearn.com/cx4sXqDpaYOU/?ref=app

How can I get the function to return more than one values?

Answer 1

Instead of printing the values, add them to a list, then return it.

def porm(a):
lst = []
k = 0
for i in range(1, a + 1 ):
for j in range(1, i + 1):
if i % j == 0:
k = k + 1
if k == 2:
lst.append(i)
k = 0
return lst

print(porm(42))
# [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41]

Answer 2

ohh I didnt think of that. Thank you very much