python - 使用 numpy 数组的维度作为 if 语句 python

Question

我将遍历不同的 numpy 数组。有些是二维的，有些是一维的。如果评估的数组是一维的，我会想把它平铺成一个二维数组。示例:

c = {'a': np.array([1, 2, 3]), 'b' : np.array([[1, 2, 3], [4, 5, 6]])}

for k in c:
    if c[k].shape #is 1D:
        c[k] = np.tile(c[k], (len(c[k]),1))

我不知道如何运行该条件。有任何想法吗？我试过类似的东西

aa = np.array([1, 2, 3])
aa.shape[0]
# 3
aa.shape[1]
# Gives an out of range error

我想通过发现没有第二维可以发现它是一维数组。但我不知道如何在 if 语句中对此进行编码。

谢谢

Answer 1

NumPy 数组有一个名为 ndim 的属性，它准确地代表了您认为它所做的事情:数组的维数。所以，你可以这样做:

if c[k].ndim == 1:
    # do something