작업형1-2<ul><li>s1, s2, s3, s4, s5, s6, 컬럼의 행(row)별 합을 구하고 그 합이 0.1보다 큰 값의 수를 구하시오</li></ul>풀이를 보면 문제에서 요구하는 컬럼명을 아래와 같이 수기로 입력하여 cols에 대입하였는데요.<ul><li>cols = ['s1', 's2', 's3', 's4', 's5', 's6']</li></ul>수기로 입력하지 않고 컬럼명을 추출하는 방법이 있는지 궁금합니다!

컬럼명을 확인해보면 뒤에 6개 컬럼입니다.<pre class="hljs"><code>cols = df.columns[-6:].tolist()</code></pre>이와 같이 뒤에서 6개의 컬럼명을 가져와 리스트 형태로 만드는 방법이 있어요:) 이러한 방식을 사용할 때는 print(cols)로 원하는 컬럼명이 들어갔는지 반드시 확인바랍니다 !!

인프런 커뮤니티 질문&답변

작업형1 모의고사 풀어보기 (columns명 추출)