numpy 배열에 .csv 파일을 열 벡터로로드하십시오.

Question

numpy 배열에 열 벡터로로드하려고하는 .csv 파일이 있습니다.

[array([[0], 
     [0], 
     [0], 
     ... 
     [0], 
     [1], 
     [0], 
     ... 
     [0], 
     [0]]), array([[0], 
     [0], 
     [0], 
     ... 
     [0], 
     [1], 
     [0], 
     ...)] 

내가 할 수 있어요 : 반면

[ 0. ... 0. 1. 0. ... 0.], [ 0. ... 1. 0. 0. ... 0.] 

는이 같은 일을하고 싶습니다 :

temp = np.genfromtxt(location, delimiter=',') 

를 사용하는 경우 데이터는 다음과 같은 항목이 표시되는 numpy array이다 이것을 사용하여 수행

my_data = [] 

for i in range(len(temp)): 
    foo = np.array([temp[i]]).T 
    my_data.append(foo) 

그러나 필요한 결과를 얻는보다 효율적인 방법이 있는지 궁금합니다.

Answer 1

temp에 새 축을 추가 캐스트 하위 목록을 순이익 배열에 :

temp = np.genfromtxt(location, delimiter=',') 
my_data = map(np.array, temp[:, :, np.newaxis]) 

---

파이썬 3를 들어, 위의 결과에 list를 호출하거나리스트 통합 사용

my_data = [np.array(arr) for arr in temp[:, :, np.newaxis]] 

을