from sklearn import datasets
diabetes = datasets.load_diabetes() #载入数据
print diabetes.data #数据
print diabetes.target #类标
print u'总行数: ', len(diabetes.data), len(diabetes.target) #数据总行数
print u'特征数: ', len(diabetes.data[0]) #每行数据集维数
print u'数据类型: ', diabetes.data.shape #类型
print type(diabetes.data), type(diabetes.target) #数据集类型
x_train = np.array([[1, 2, 3],
[1, 3, 4],
[2, 1, 2],
[4, 5, 6],
[3, 5, 3],
[1, 7, 2]])
print(x_train[:,np.newaxis])
print(x_train[:,np.newaxis,2])
可以查看结果