[Python] *args란?/ zip과 연관시켜 이용하기

 

 >> 💻 파이썬(Python): *args/ zip에 대하여

 

 

 

 스터디하면서 나왔던 이 코드! *arr..? 저게 뭔가 싶었다.

 찾아보니, 매개변수로 몇 개를 넣어줄지 모를 때에 사용하는 코드라고 한다.

 여러가지를 찍어보는데...찍어볼수록 아리송하고 감이 잘 안잡힌다. 많이 써봐야 할 것 같다...

 

 

 

 

 

💙 💙 💙

 

 

arr = [
    [1,2,3,4],
    [5,6,7,8],
    [1,2,3,4],
    [5,6,7,8]
]

print(*arr)
# [1,2,3,4] [5,6,7,8] [1,2,3,4] [5,6,7,8]

print(*arr[0])
# 1 2 3 4

 일반적으로 일반 배열에 *을 붙여 바로 찍어보면, 그 안의 원소들을 하나씩 돌려주는 것 같다.

 *arr은 맨 겉의 [ ] 을 풀어헤쳐서 안의 네 개의 배열을 리턴한다.

 *arr[0]의 경우 arr[0]이 [1,2,3,4]이므로 이 안의 원소들을 풀어헤쳐 1 2 3 4를 리턴한다.

 

arr = [
    [1,2,3,4],
    [5,6,7,8],
    [1,2,3,4],
    [5,6,7,8]
]

def test(*args):
    print(args)
    print(*args)
    
test(arr)
# print(args) : ([[1,2,3,4], [5,6,7,8], [1,2,3,4], [5,6,7,8]])
# print(*args): [[1,2,3,4], [5,6,7,8], [1,2,3,4], [5,6,7,8]]

test(*arr)
# print(args) : ([1,2,3,4], [5,6,7,8], [1,2,3,4], [5,6,7,8])
# print(*args): [1,2,3,4] [5,6,7,8] [1,2,3,4] [5,6,7,8]

 

 여기서부터 멘붕......

 자. 자세히 보면 짐작이 간다!

 *args의 경우 풀어헤친다!고 이해를 했는데, 이게 그냥 풀어헤칠 때와 함수의 파라미터로 전달될 때가 다르다.

 그냥 print(*args)할 때는 args가 풀어헤쳐져 그 안의 원소들이 나타나지만,

 함수의 파라미터로 전달될 때는 풀어헤쳐진 원소들이 튜플로 묶이게 된다!

 함수 안에서 프린트로 *을 쓰면, 그 바깥의 튜플이 벗겨져서 프린트된다.

 

 따라서 위 코드에서 test(arr)의 경우에는, arr 큰 배열 '전체'가 *args로 인해 파라미터로 들어오는 여러 인자들 중 '하나'로 인식되기 때문에, 큰 튜플 안에 arr이 통째로 하나의 원소로써 들어가 있는 것이다. 

 만약 test(arr, [0,0,0])이 된다면 print(args)의 출력은 ([[1,2,3,4], [5,6,7,8], [1,2,3,4], [5,6,7,8]], [0,0,0])이 될 것이다.

 

 test(*arr)의 경우는 arr을 한 번 풀어헤친 안쪽의 네 개의 배열들이 *args가 받는 여러 인자가 되어 네 개의 원소로써 하나의 튜플 안에 들어가 있다.

 

 

 

 

arr = [
    [1,2,3,4],
    [5,6,7,8],
    [1,2,3,4],
    [5,6,7,8]
]

for col in zip(*arr):
    print(col)
   
# (1, 5, 1, 5)
# (2, 6, 2, 6)
# (3, 7, 3, 7)
# (4, 8, 4, 8)

print(list(zip(*arr)))
#[(1, 5, 1, 5), (2, 6, 2, 6), (3, 7, 3, 7), (4, 8, 4, 8)]

 

 위까지 이해하고 나면 이제 zip을 이용해서 어떻게 열만 뽑아낼 수 있는지 이해가 간다!

 *arr로 제일 겉 괄호[]를 벗긴 다음, 남은 안쪽 원소, 즉 네 개의 배열을 zip()함수에 넣어준다.

 zip은 리스트나 튜플과 같은 자료형에서 같은 위치에 있는 것들끼리 튜플로 묶어주는 역할을 하므로, 포장이 벗겨지고 남은 네 개의 배열에서 위치가 같은(열이 같은) 값들이 튜플로 묶여 프린트된다.

 

 

 

 

이렇게 파이썬에 한 발짝 가까워진다...

 

 

 

 

 

 

 

 

💙 💙 💙

 

 

Ｐｏｓｔｅｄ Ｂｙ ＳＡＹ

𝘛𝘩𝘢𝘯𝘬𝘴 𝘧𝘰𝘳 𝘳𝘦𝘢𝘥𝘪𝘯𝘨