[Python] 리스트 조작

- 리스트에 요소 추가

 

1. append

 

>>> a = [10, 20, 30, 40]
>>> a.append(50)
>>> a
[10, 20, 30, 40, 50]
>>> a.append(30, 50)
Traceback (most recent call last):
  File "<pyshell#125>", line 1, in <module>
    a.append(30, 50)
TypeError: list.append() takes exactly one argument (2 given)

 

 

리스트명.append(요소)를 사용하여 리스트의 요소 끝에 요소 하나를 추가할 수 있습니다. 오직 요소 하나만 추가할 수 있기에 요소 두개를 넣으면 위처럼 에러가 발생합니다.

 

2. extend

 

>>> a = [10, 20, 30, 40]
>>> a.extend([50, 60])
>>> a
[10, 20, 30, 40, 50, 60]

 

그렇다면 리스트에 하나 이상의 요소를 추가하고 싶다면 어떻게 해야 할까요? 바로 리스트명.extend([요소1, 요소2])를 사용하면됩니다. 하나 이상의 리스트가 리스트 a 안으로 들어간 것을 볼 수 있습니다.

 

3. insert

 

a = [10, 20, 30, 40]
>>> a.insert(2, '안녕')
>>> a
[10, 20, '안녕', 30, 40]

 

인덱스를 지정해 원하는 위치에 요소를 추가하고 싶다면 리스트명.insert(인덱스, 요소)를 사용하면 됩니다. 인덱스 2를 지정하고 요소 '안녕'을 넣으니 인덱스 2의 위치에 요소 '안녕'이 들어간 것을 볼 수 있습니다.

 

 

- 리스트에 요소 삭제

리스트에 요소를 추가할 수도 있다면 삭제할 수도 있습니다.

 

1. pop

 

>>> a = [10, 20, 30, 40]
>>> a.pop()
40
>>> a
[10, 20, 30]

 

요소.pop()을 사용하면 리스트 맨 끝의 요소를 반환하고 삭제합니다. 맨끝 요소 40이 삭제된 것을 볼 수 있습니다.

 

>>> a = [10, 20, 30, 40]
>>> a.pop(1)
20
>>> a
[10, 30, 40]

 

괄호 안에 인덱스를 지정해 원하는 위치의 요소를 삭제할 수도 있습니다.

 

2. remove

 

pop은 인덱스를 지정해 요소를 삭제했자면 원하는 요소를 지정해 삭제할 수도 있습니다.

 

>>> a = [10, 20, 30, 40]
>>> a.remove(20)
>>> a
[10, 30, 40]

 

괄호 안에 넣은 요소가 삭제된 것을 볼 수 있습니다.

 

>>> a = [10, 20, 30, 40, 20]
>>> a.remove(20)
>>> a
[10, 30, 40, 20]

 

만약 리스트 안에 똑같은 요소가 2개 있다면 먼저 위치한 요소를 삭제합니다.