파이썬 CSV 파일을 numpy 배열로, 첫 번째 행을 문자열로, 나머지는 부동 소수점으로 읽습니다.

첫 번째 행이 문자열 (열 이름)이고 나머지 행이 숫자 인 CSV에 데이터가 저장되어 있습니다. 이것을 numpy 배열에 어떻게 저장합니까? 내가 찾을 수있는 것은 행이 아닌 열에 대한 데이터 유형을 설정하는 방법입니다.

지금은 계산을 수행하기 위해 헤더를 건너 뛰고 있지만 최종 버전에 헤더가 있어야합니다. 그러나 헤더를 그대로두면 전체 배열이 문자열로 설정되고 계산이 실패합니다.

이것이 내가 가진 것입니다.

 data = np.genfromtxt(path_to_csv, dtype=None, delimiter=',', skip_header=1) 

 

해결 방법

 

np.genfromtxt 함수에서 names = True 인수를 사용하면 열 이름을 유지할 수 있습니다.

 data = np.genfromtxt(path_to_csv, dtype=float, delimiter=',', names=True) 

데이터를 float로 변환하는 dtype = float 에 유의하세요. 이는 np.genfromtxt 에 데이터 유형을 추측하도록 요청하는 dtype = None 을 사용하는 것보다 더 효율적입니다.

 

참조 페이지 https://stackoverflow.com/questions/12336234