파이썬 NumPy loadtxt 데이터 유형

다음과 같은 데이터 세트를로드하려고합니다.

Algeria,73.131000,6406.8166213983,0.1
Angola,51.093000,5519.1831786593,2
Argentina,75.901000,15741.0457726686,0.5
Armenia,74.241000,4748.9285847709,0.1

마지막에는 열 1과 2 만 필요합니다. 국가 이름과 마지막 열은 필요하지 않습니다. 기본적으로 차원이 nx1 인 두 개의 행렬을 추출해야합니다. 데이터 유형을 지정해야한다는 것을 알고 있습니다.

data=np.loadtxt('file.txt',delimiter=',',dtype=[('f0',str),('f1',float),('f2',float),('f3',float)])

그러나 이것은 튜플 목록을 생성합니다.

array([('', 73.131, 6406.8166213983, 0.1),
   ('', 51.093, 5519.1831786593, 2.0),`

대신에

array(['',73.131,6406.8166213983,0.1],
      ['',51.093, 5519.1831786593, 2.0],

실수는 어디에 있습니까?

 

해결 방법

 

import numpy as np
col1 = np.genfromtxt('yourfile.txt',usecols=(1),delimiter=',',dtype=None)
col2 = np.genfromtxt('yourfile.txt',usecols=(2),delimiter=',',dtype=None)

또는 둘 다 함께 :

np.genfromtxt('yourfile.txt',usecols=(1,2),delimiter=',',dtype=None)

 

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