문제를 풀다가 리스트에서 한 번에 모든 원소에서 같은 값을 뺄 수 있는 방법이 있는지 찾아보다가 발견하여 기록해둔다.
[1] map 이용
가장 익숙한 map과 lambda를 이용하는 방법이다. 원래 저장하고 있던 리스트의 각 원소에서 어떤 값을 어떻게 처리할 것인지 lambda의 조건에 작성하면 된다.
array2 = list(map(lambda x: x-n, array1))
[2] List Comprehension 이용
array2 = [i-1 for i in array1]
+ numpy 이용: numpy 설치 후 numpy 내의 subtract 함수를 사용한다.
사용 방법은 아래와 같다.
array2 = numpy.subtract(array1, 1)
참고
https://jlim0316.tistory.com/306
'코딩테스트 > 스터디' 카테고리의 다른 글
[코테 스터디] sys.stdin.readline()으로 입력 받기 (0) | 2024.03.23 |
---|---|
[코테 스터디] Week4 이진 탐색 개념 정리 (1) | 2024.03.23 |
[코테 스터디] Week3 정렬 개념 정리 (0) | 2024.03.23 |
[코테 스터디] Week2 DFS & BFS 개념 정리 (4) | 2024.03.14 |
[코테 스터디] Week1 그리디 & 구현 개념 정리 (3) | 2024.03.09 |