본문 바로가기
Python

Python 리스트의 원소들을 2개씩 묶어서 합 구하기

by jayden-lee 2019. 5. 11.
728x90

자연수 N을 입력 받아서 N개의 1 이상 5 이하의 난수로 이루어진 리스트 a를 생성한다. 생성된 리스트의 원소들을 2개씩 묶어서 합을 구한 다음 새로운 리스트 b를 구하라.

import random

def twoSum(a, n):
    b = []
    for i in range(0, n, 2):
        if (i == n-1):
            b.append(a[i])
        else:
            b.append(a[i] + a[i+1])

    return b;


N = int(input('N = '))

a = []

# N 개수 만큼 1 이상 5 이하 난수로 이루어진 숫자 생성
for i in range(N):
    a.append(random.randrange(1, 6))

print('a =', a)
print('b =', twoSum(a, N))

 

댓글