Python 按索引查找值

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

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

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

这里y是在b中存储非Nan值索引的列。

---

我有另一列x，必须将b与x的值匹配。过滤掉这些值并将其存储在另一列中。

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

df[x].iloc[b]

您可以使用values属性获取值

df[x].iloc[b].values

---

或者，如果需要列表，请执行以下操作：

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

---

您正在使用pandas数据帧吗？请使用

df.dropna（inplace=True）

并执行您的操作。根据您的代码，b应存储y的非空值索引，但您的问题说明了其他情况。