Как найти все индексы в 2-D матрице, где кадр данных pandas имеет значения не-NA

Question

Я использую большой 2-D pandas dataframe для алгоритма совместной фильтрации и хочу создать список списков, который предоставляет индексы всех не-NA значений из 2-D pandas Dataframe.

2-D датафрейм выглядит так:

    0        1        2
0   1        NA       2
1   NA       5        NA
2   NA       NA       NA
3   1        3        4
4   4        NA       1

Ниже приведен список ожидаемых выходных данных (я допустил ошибку при упоминании ожидаемого вывода в исходной публикации и исправилэто):

[[0,0], [0,2], [1,1], [3,0], [3,1], [3,2], [4,0], [4,2]]

Answer 1

Ваш ожидаемый вывод содержит индексы, где - это NaN значения.

Чтобы получить их, используйте: np.argwhere(np.isnan(df.values)).tolist()

Или, если вы хотите, чтобы индексы не NaN элементов, использование:

np.argwhere(~np.isnan(df.values)).tolist()