문제

문제 7033

행 슬라이싱하기

시간 제한 1.000초 메모리 제한 128MB

문제 설명

하나의 리스트(l)와 정수(n)이 주어졌을 때, 리스트에서 n행의 값을 슬리이싱하시오.

입력 설명

첫 번째 줄에는 리스트의 크기 (r, c)가 공백으로 구분되어 입력된다.
두 번째 줄에서부터 r+1번째 줄까지 각 줄에는 행 데이터가 공백으로 구분되어 입력된다.
r+2 번째 줄에는 정수 n이 입력된다.

출력 설명

슬라이싱 된 행이 넘파이 배열로 출력된다.

입력 예시

2 3
1 2 3
4 5 6
1

출력 예시

[4 5 6]

출처

Numpy