python - 从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

是的，它将返回形状为 (57,) 的数组。如果你想变成 (57,1) ，你可以通过 reshape() 来实现。

col1=(result[:, 1]).reshape(-1,1)