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]]
2
6
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐