如何选择具有多个索引的数据？

Question

我有以下设置：

<class 'pandas.core.series.Series'>类型的数据集1：

Attr1
12
7
8

<class 'numpy.ndarray'>类型的Dataset2-data-probability

0.001
0.5
0.0001

我的目标是从数据集1中选择概率小于0.02的值。实际上，数据集1和数据集2具有相同的基数。 我的尝试如下：

1. 定义条件：my_preferred_data = np.asarray(np.where(Dataset2 < 0.02))
2. 以dataset1[my_preferred_data]的身份获取数据

我遇到的错误是ValueError: Cannot index with multidimensional key. 请，我可以帮忙吗？非常感谢。

Answer 1

您可以直接这样做：

Dataset1[Dataset2 < 0.02]

这会给你答案。