Как преобразовать DataFrame в массив целых чисел, не используя при этом значения nan?

Question

У меня есть следующее DataFrame с 3 столбцами.

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

Мне нужно преобразовать это в массив целых чисел и пропустить значения 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]]