sklearn 自带数据集属性

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])

可以查看结果