Python 3.x 如何在数据帧具有非NA值的二维矩阵中查找所有索引

Python 3.x 如何在数据帧具有非NA值的二维矩阵中查找所有索引,python-3.x,pandas,dataframe,Python 3.x,Pandas,Dataframe,我正在使用一个大的2-Dpandas数据帧进行协同过滤算法,并希望创建一个列表列表,该列表提供来自2-D pandas数据帧的所有非NA值的索引 二维数据帧类似于: 0 1 2 0 1 NA 2 1 NA 5 NA 2 NA NA NA 3 1 3 4 4 4 NA 1 预期输出列表如下(我在原区域帖子中提到预

我正在使用一个大的2-Dpandas数据帧进行协同过滤算法,并希望创建一个列表列表,该列表提供来自2-D pandas数据帧的所有非NA值的索引

二维数据帧类似于:

    0        1        2
0   1        NA       2
1   NA       5        NA
2   NA       NA       NA
3   1        3        4
4   4        NA       1
预期输出列表如下(我在原区域帖子中提到预期输出时犯了一个错误,并已更正):


您的预期输出包含有值的索引

要获取它们,请使用:
np.argwhere(np.isnan(df.values)).tolist()

或者,如果您想要而不是NaN元素的索引,请使用:

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

我在预期输出列表中犯了一个错误,我已经纠正了它
np.argwhere(~np.isnan(df.values)).tolist()