我先做一个小例子热身一下:
import numpy as np
dataset=np.array([[1.,-1.,2.],[2.,0.,0.],[0.,1.,-1.]])
X=dataset[0]#[ 1. -1. 2.]第一行
X=dataset[:1]#[[ 1. -1. 2.]]第一行很少用
X=dataset[:,1]#[-1. 0. 1.]第二列
X=dataset[:,0:2]
# =============================================================================
# [[ 1. -1.]
# [ 2. 0.]
# [ 0. 1.]]
# =========================================================