明智地寻找价值指数

Question

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

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

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

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

Answer 1

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

df[x].iloc[b]

您可以使用values属性来获取值

df[x].iloc[b].values

Answer 2

或者如果想要列表做的话：

print(df[column].iloc[b].values.tolist())