要获取numpy数组的行数和列数,可以使用shape属性。可以通过shape属性获得一个元组,元组的第一个元素表示数组的行数,第二个元素表示数组的列数。以下是一个示例:
import numpy as np # 创建一个2维数组 arr = np.array([[1, 2, 3], [4, 5, 6]]) # 获取数组的行数和列数 rows, cols = arr.shape print("行数:", rows) print("列数:", cols)
运行上述代码,将输出以下结果:
行数: 2 列数: 3