根据另一个数组中的值从numpy中选择列

Question

我有一个数据集，X形状（1000,1024）和一个数组，排名形状（1024，）。 rank数组为每列保留一个值，其中1列出现823次。我想从X中选择排名为1的列。预期输出的形状应为（1000,823）。我试过这个：

Aa = np.where(X[:,rank ==1])

但输出错误。请帮助！