[TIL / Python] 2차원 배열 슬라이싱, sum()

프로그래머스 문제를 풀다가 2차원 배열을 슬라이싱 할때,

그냥 arr[:n][:n] 처럼 하면 될 줄 알았는데 [[1,1],[1,0]] 처럼 될 줄 알았는데 [[1,1,0,0],[1,0,0,0]] 이렇게 출력되었다. 또한, 2차원 배열은 1차원 배열처럼 sum(li1)이 되지않아 번거러웠는데 이를 쉽게 해결하는 방법으로 import numpy as np 가 있다.

✅ import numpy as np

• a = np.array([2,3,4])
  -> np 생성
  Tuple이나 List를 numpy.ndarray로 만듬

• .sum(): 모든 요소의 합
  2차원 배열의 합도 가능

• .min(): 모든 요소 중 최소값

• .max(): 모든 요소 중 최대값

• .argmax(): 모든 요소 중 최대값의 인덱스

• .cumsum(): 모든 요소의 누적합

• a = np.arange(15).reshape(3, 5)

  a= np.arange(15).reshape(3, 5)
  print(a)
  
  # [[ 0  1  2  3  4]
  #  [ 5  6  7  8  9]
  #  [10 11 12 13 14]]