-
카테고리
-
세부 분야
자격증 (데이터 사이언스)
-
해결 여부
해결됨
작업형1 모의고사 풀어보기 (columns명 추출)
24.05.20 11:14 작성 조회수 72
0
작업형1-2
s1, s2, s3, s4, s5, s6, 컬럼의 행(row)별 합을 구하고 그 합이 0.1보다 큰 값의 수를 구하시오
풀이를 보면 문제에서 요구하는 컬럼명을 아래와 같이 수기로 입력하여 cols에 대입하였는데요.
cols = ['s1', 's2', 's3', 's4', 's5', 's6']
수기로 입력하지 않고 컬럼명을 추출하는 방법이 있는지 궁금합니다!
답변을 작성해보세요.
0
퇴근후딴짓
지식공유자2024.05.20
컬럼명을 확인해보면 뒤에 6개 컬럼입니다.
cols = df.columns[-6:].tolist()
이와 같이 뒤에서 6개의 컬럼명을 가져와 리스트 형태로 만드는 방법이 있어요:)
이러한 방식을 사용할 때는 print(cols)로 원하는 컬럼명이 들어갔는지 반드시 확인바랍니다 !!
답변 1