首页 > 解决方案 > 如何检索 numpy.array 的维度

问题描述

我对学习 python 非常陌生,我正在尝试在np. array nxm中缩放矩阵。问题是,如果将 中的矩阵np.array作为输入给出并且我不知道矩阵的范围有多大,我如何初始化大小m?Python中是否有某些features可以用于此目的?

import numpy as np
def scaleArray(arr: np.array);
    ##how ??

arrayB = np.array([[1,2,4],
                   [3,4,5],
                   [2,1,0],
                   [0,1,0]])
scaleArray(b)

arrayB只是例如

预期输出:

3

标签: pythonarraysnumpymatrix

解决方案


arr.shape是您要查找的内容,它为您提供了 nD 数组的尺寸。

在你的情况下,你想要arr.shape[1]


推荐阅读