xtensor：具有特定列值的 Select 行

Question

我正在玩xtensor ，我只想對具有特定列值的 select 行執行一個簡單的操作。 想象一下我有以下數組。

[ 
  [0, 1, 1, 3, 4 ]
  [0, 2, 1, 5, 6 ]
  [0, 3, 1, 3, 2 ]
  [0, 4, 1, 5, 7 ]
]

現在我想 select col2和col4值為 3 的行。在這種情況下是row 3 。

  [0, 3, 1, 3, 2 ]

我想達到類似於這個答案所達到的效果。

我怎樣才能在xtensor中實現這一點？

Answer 1

go 的方法是用您需要的列進行切片，然后查看所有列的條件在哪里成立。

對於后者， xt::all(...)的重載似乎尚未實現（尚未實現），但我們可以使用xt::sum(..., axis)來實現相同的目的：

#include <xtensor/xtensor.hpp>
#include <xtensor/xview.hpp>
#include <xtensor/xio.hpp>

int main()
{
  xt::xtensor<int,2> a =
    {{0, 1, 1, 3, 4},
     {0, 2, 1, 5, 6},
     {0, 3, 1, 3, 2},
     {0, 4, 1, 5, 7}};

  auto test = xt::equal(xt::view(a, xt::all(), xt::keep(1, 3)), 3);
  auto n = xt::sum(test, 1);
  auto idx = xt::flatten_indices(xt::argwhere(xt::equal(n, 2)));

  auto b = xt::view(a, xt::keep(idx), xt::all());

  std::cout << b << std::endl;

  return 0;
}