Question

我正在学习c ++，我想在c ++中实现以下python代码：

C = np.where(A>B)[0]
while len(C)>0:
    d = C[0]
    # do something to A[d] and B[d]
    C = A>B

A和B都是相同长度的向量。在C ++中，我知道如何使用A声明和初始化B和vector，并为A和B实现中间“做一些部分”，但我不知道如何比较它们并检查A是否包含大于B的元素，并找到发生这种情况的元素的索引。

Answer 1

C ++在<algorithm>标题中有一组丰富的实用函数。如果您有问题：

C = np.where(A>B)[0]可以按如下方式翻译成C ++：

std::size_t index = 0;
auto pos = std::find_if(A.cbegin(), A.cend(), [&index, &B](const int &i){
    return i > B[index++];
});

C = A>B也可以在C ++中重写如下：

std::size_t index = 0;
auto is_okay = std::all_of(A.cbegin(), A.cend(), [&index, &B](const int &i){
    return i > B[index++];
});

因此，它可以简化如下：

std::vector<int> A = {/* contents of A */};
std::vector<int> B = {/* contents of B */};

std::size_t index;
auto greaterThanB = [&index, &B](const int &i){
    return i > B[index++];
};

// C = np.where(A>B)[0]
index = 0;
auto pos = std::find_if(A.cbegin(), A.cend(), greaterThanB);

// C = A>B
index = 0;
auto is_okay = std::all_of(A.cbegin(), A.cend(), greaterThanB);

另请注意，此代码pos的类型为vector<int>::iterator，指向第一个匹配项。要将其转换为整数索引，可以使用std::distance函数。

std::size_t index = std::distance(A.cbegin(), pos);

在c ++中找到满足给定条件的向量元素的位置

1 个答案: