Python NumPy：将只有一列的数据从csv导入2D数组

Question

我正在尝试将此数据从csv文件导入2D数组：

0
1
1
0

这是我的代码：

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

我正在寻找这样的结构：

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

但是它出现了这样：

[ 0.  1.  1.  0.]

只要我向数据添加第二列，它就会以正确的方式出现。我正在寻找一个通用的解决方案，所以我不必为一列数据编写任何特殊情况。

谢谢：）

Answer 1

尝试

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