Question

我有两个数组y dims (1x100)和X dims (3x100)

我想从X中选择y为{0}的所有值以及y为1的所有值。

我尝试X[y==0]来获取y为0的所有值，但它不适用于X的dim 0，即3。

有关如何操作的任何建议吗？

由于

Answer 1

您需要正确对齐尺寸

X[:, y[0] == 0]

第一个:指定抓取第一维中的所有内容。在第二个插槽中使用y[0] == 0指定我们正在索引第二个维度。