문제를 풀다가 리스트에서 한 번에 모든 원소에서 같은 값을 뺄 수 있는 방법이 있는지 찾아보다가 발견하여 기록해둔다.
[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
[Python] 리스트에서 같은 값을 빼는 법
1. numpy 이용 1 2 3 4 a = [1,2,3] output = np.subtract(a, 1) print(output) [0 1 2] 2.List Comprehension 이용 1 2 3 4 a = [1,2,3] output1 = [i-1 for i in a] print(output1) # [0, 1, 2] 3. map 이용 1 2 3 output2 = list(map(lambda x:x -1, a)) print(outpu
jlim0316.tistory.com
'코딩테스트 > 스터디' 카테고리의 다른 글
[코테 스터디] 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 |