numpy vectorize在（2，）数组上使用

Question

我有一个（m，2）的数组数组，我想使用以下功能将其转换为（m，1）的形状。

def func(x):
    if x == [1., 1.]:
        return 0.
    if x == [-1., 1.] or x == [-1., -1.]:
        return 1.
    if x == [1., -1.]:
        return 2.

我希望将其应用于（M，2）内的每个（2，）向量，数组导致（M，1） array。我尝试使用numpy.Dectorize，但是似乎该函数在数组的每个元素中都应用（在通常的情况下是有意义的）。所以我没有应用它。

我的目的不是用于循环。谁能帮我吗？谢谢。

Answer 1

import numpy as np


def f(a, b):
    return a + b


F = np.vectorize(f)
x = np.asarray([[1, 2], [3, 4], [5, 6]]).T

print(F(*x))

输出：

[3, 7, 11]