python - 如何将 DataFrame 转换为整数数组，同时省略 nan 值？

Question

我有以下包含 3 列的 DataFrame。

A  B  C
-------
1  2  4
2  3  nan
4  4  2

我需要将其转换为 numpy 整数数组并省略 nan 值。

所以我需要这个:

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

我该怎么做？我特别需要结果数组中的元素为整数并省略 nan 值。

Answer 1

您可以使用列表理解:

In [11]: [[int(i) for i in r if not np.isnan(i)] for _, r in df.iterrows()]
Out[11]: [[1, 2, 4], [2, 3], [4, 4, 2]]

In [12]: [[int(i) for i in r if not np.isnan(i)] for _, r in df.iteritems()]
Out[12]: [[1, 2, 4], [2, 3, 4], [4, 2]]