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

Question

我下面的DataFrame有3列。

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]]