将数组拆分为预测矩阵和响应向量

Question

我认为这是一个非常微不足道的问题，但希望有人可以帮助我。拆分包含预测变量（输入）和响应变量（输出）的数组的最佳方法是什么？

我导入了一个包含预测变量和响应的csv文件，但是我想将其拆分以使预测变量处于一个numpy矩阵（n X m）并且响应位于一个向量中（n X 1 ）。有什么想法吗？

感谢。

Answer 1

只需使用导入的numpy数组的shape属性来确定m并使用切片提取子数组：

import numpy as np

# load csv data
data = np.loadtxt('data.txt', delimiter=',')
# m is the number of columns minus one
m = data.shape[1]-1
# use slicing to extract subarrays
pred = data[:,:m]
resp = data[:,m]

可以找到切片语法的一个很好的解释here。