2017-04-03 29 views
2

我想從一個CSV文件裏的數據導入到一個二維數組:的Python NumPy的:只有一列從CSV導入數據到二維數組

0 
1 
1 
0 

這是我的代碼:

Y = np.genfromtxt("output_data.csv", delimiter=",") 

我在尋找這樣的結構:

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

而是它出來是這樣的:

[ 0. 1. 1. 0.] 

只要向數據添加第二列,它就會以正確的方式出現。我正在尋找一個通用的解決方案,因此我不必爲單個數據列編寫任何特殊情況。

謝謝:)

回答

4

嘗試

Y = np.loadtxt("output_data.csv", delimiter=",", ndmin=2) 
+0

這工作。非常感謝! –