从python中的矩阵中提取列作为向量

Question

我有一个csv文件，正在使用以下命令将其转换为矩阵：

reader = csv.reader(open("spambase_X.csv", "r"), delimiter=",")
x = list(reader)
result = numpy.array(x)
print(result.shape)   #outputs (57,4601)

现在我要提取矩阵结果的第一列，我将通过以下操作进行提取：

col1=(result[:, 1])

**print(col1.shape)        #outputs (57,)**

为什么不打印为（57,1）。我该怎么办？

TIA

Answer 1

您可以添加[]

result[:,[1]].shape
Out[284]: (2, 1)

---

数据输入

result
Out[285]: 
array([[1, 2, 3],
       [1, 2, 3]])

---

更多信息

result[:,[1]]
Out[286]: 
array([[2],
       [2]])
result[:,1]
Out[287]: array([2, 2])