python - 尝试通过仅获取元组中的整数来形成一个数组，该数组从元组解包中获取其大小

Question

if (type(n) == int):
    y = np.zeros(n)
    y[0] = seed
elif (type(n) == tuple):
    for a,b in n:
        if (type(a) == int):
            y = np.zeros(a)
            y[0] = seed
        elif (type(b) == int):
            y = np.zeros(b)
            y[0] = seed

我一直试图通过元组解包来获取此代码，以仅获取输入到元组中的整数，并使用该整数形​​成一个充满零的数组，并使用该整数形​​成其大小并丢弃其余部分，但是我似乎无法避免收到错误消息TypeError: 'int' object is not iterable 无论如何你可以帮我解决这个问题吗，这段代码中有两种情况，其中 n 可以是整数或元组。感谢您的帮助

Answer 1

看起来n是一个(二元素？)元组 - 比如说，(x,y)。那么 for a,b in n: 行本质上等同于 for a,b in (x,y):。在循环的第一次迭代中，Python 尝试将元组 x 的第一个元素分配给 a,b，这是不可能的。

您应该用赋值a,b=n替换循环，或者只使用n[0]和n[1]而不是 a 和 b。