리스트 2

[파이썬] csv 파일 가져와서 데이터를 리스트에 담기

파이썬에서 csv에 있는 데이터를 사용하려면 csv 파일에 있는 데이터를 우선 2차원 리스트로 담는 것이 편하다. 다음과 같이 코드를 구성하면 간단하게 데이터를 가져올 수 있다. 5번 행의 open 함수 부분에서 1번째 인자로 csv 파일 경로명과 2번째 인자인 인코딩 방식만 변경해주고 나머지는 그대로 써도 된다. 7번 행은 csv 파일에서 1번째 행인 컬럼 명들을 제외시키고 2번째 행부터 데이터들을 가져올 수 있다. 컬럼 명들도 포함시키고 싶으면 7번 행을 주석 처리 하면 된다. 파일을 다 사용했으면 항상 close() 함수로 파일을 닫아야 한다. 그렇지 않으면 다른 코드에서 똑같은 파일을 사용할 경우 에러가 날 수 있기 때문이다. 비슷한 예를 들면 우리가 평소에 한글이나 엑셀 파일을 실행한 상태로 그..

[파이썬] 폴더에 원하는 파일명을 가진 파일 찾기

폴더에 원하는 파일을 가져와서 사용하고 싶을때 활용할 수 있는 코드이다. 우선 os.listdir()을 활용해서 폴더에 있는 파일들을 리스트로 저장한 다음에 특정 파일명이 포함되어 있으면 리스트로 저장하는 코드로 해결했다. 실행 결과로 test.py, readCsv.py가 출력되는 것을 확인했다. 복붙 코드 1 2 3 4 5 6 7 8 import os.path files_list = os.listdir('./') # 폴더에 있는 파일들을 리스트로 저장 # 파일명에 test 라는 이름이 포함된 파일이 있으면 저장 file_name = [s for s in files_list if "test" in s] # 파일명에 read 라는 이름이 포함된 파일이 있으면 저장 file_name = [s for s in..