• 카테고리

    질문 & 답변
  • 세부 분야

    자격증 (데이터 사이언스)

  • 해결 여부

    해결됨

작업형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']

수기로 입력하지 않고 컬럼명을 추출하는 방법이 있는지 궁금합니다!

답변 1

답변을 작성해보세요.

0

컬럼명을 확인해보면 뒤에 6개 컬럼입니다.

cols = df.columns[-6:].tolist()

이와 같이 뒤에서 6개의 컬럼명을 가져와 리스트 형태로 만드는 방법이 있어요:)

이러한 방식을 사용할 때는 print(cols)로 원하는 컬럼명이 들어갔는지 반드시 확인바랍니다 !!

지현님의 프로필

지현

질문자

2024.05.20

감사합니다!! ^^