Поиск значений по индексу

Question

Я должен выполнить RMSE для двух столбцов с разными значениями не-Nan. Я нашел индексы значений Non-Nan в первом столбце. Теперь я отфильтровал значения 2-го столбца по этим показателям.

Это код, который я использовал для поиска значений индексов: -

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

Здесь y - столбец, в котором хранятся индексы значений Non-Nan в b. У меня есть другой столбец х и должен соответствовать б со значениями х. Отфильтруйте эти значения и сохраните в другом столбце.

Answer 1

ИЛИ, если требуется список, сделайте:

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

Answer 2

Если вы используете кадры данных pandas, вы можете использовать pandas iloc

df[x].iloc[b]

Вы можете просто получить значения, используя атрибут values ​​

df[x].iloc[b].values