이 비단뱀 코드를 개선하는 방법은 무엇입니까?

Question

나는 이것이 내가 내 코드에 대한 답을 얻고있다하지만 난 내 코드에 만족하지 오전

>>> getNumbers(10) 
[100, 64, 36, 16, 4, 0, 4, 16, 36, 64, 100] 
>>> getNumbers(9) 
[81, 49, 25, 9, 1, 1, 9, 25, 49, 81] 

을 problem-에 대해 내가 코딩해야하는 문제를하고있는 중이 야,이 코드를 개선하기 위해 몇 가지 옵션을 제안 해주십시오.

def getNumbers(num): 
myList=[] 
mylist=[] 
if num%2==0: 
    for numbers in range(num,-2,-2): 
     myList.append(numbers**2) 
    for numbers in range(2,num+2,2): 
     mylist.append(numbers**2) 

    print myList+mylist 

elif num%3==0: 
    for numbers in range(num,-1,-2): 
     myList.append(numbers**2) 
    for numbers in range(1,num+2,2): 
     mylist.append(numbers**2) 

    print myList+mylist 

else: 
    print(mylist) 

4 for loops !!! 이것은 나를 괴롭히는 것이다! !!

Answer 1

좋아요?

def getNumbers(n): 
    return [i * i for i in range(-n, n + 1, 2)]