python - 查找值索引明智

Question

我必须对具有不同 Non-Nan 值的两列执行 RMSE。我在第一列中找到了非南值的索引。现在我已经根据这些索引过滤掉了第二列的值。

这是我用来查找索引值的代码：-

b = np.argwhere(y.notnull().values).tolist()

这里 y 是在 b 中存储 Non-Nan 值的索引的列。我有另一列 x，必须将 b 与 x 的值匹配。过滤掉这些值并将其存储在另一列中。

标签： pythonnumpy

Answer 1

如果您使用的是 pandas 数据框，则可以使用 pandas iloc

df[x].iloc[b]

您可以使用 values 属性获取值

df[x].iloc[b].values