np.shape()的用法

np.shape(x)：
其中：x的取值为：0—表示行数、1—表示列数

import numpy as np
data = np.arange(12).reshape(2,6) # 生成数据并重组成2行六列
print(data)
cols = data.shape[0] # 0表示行数
print(cols)
cols1 = data.shape[1] # 1表示列数
print(cols1)  

输出结果：

[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]]