Python:配列を保ちつつnanだけを削除したい

Question

前提・実現したいこと

Pythonを使用しています。

• 実現したい内容 …

「　aの中から行列を保ちつつnanだけを削除したいです」

実行したコードとその結果

実行したコード

このようなコードを実行しました。

In[]:
a = np.array([[ nan,  nan], [ nan,  nan], [ nan,  nan], [ nan,  nan], [ nan,  nan], [ nan,  nan], [ nan,  nan], [ nan,  nan], [ 0.4,  0.5], [-0.5, -0.4], [ 0.2, -0.2], [ 0. , -0.3], [-0.4,  0.5], [ 0.5, -0.9], [ 1.5,  0.8]])

print(a.T[~np.isnan(a.T)])

実行した結果

上記のコードを実行した結果、
以下のようになりました。

Out[]:
array([ 0.4, -0.5,  0.2,  0. , -0.4,  0.5,  1.5,  0.5, -0.4, -0.2, -0.3, 0.5, -0.9,  0.8])

本来得たい結果

望んでいた実行結果は以下の通りです。

Out[]:
array([[0.4, -0.5,  0.2, 0. , -0.4,  0.5,  1.5], [0.5, -0.4, -0.2, -0.3,  0.5, -0.9,  0.8]])

Answer 1

こんなのは如何でしょう。

a[~np.all(np.isnan(a), axis=1)].T

Comments

[TeRa_130]

ありがとうございます。
加え手で申し訳ないのですが、例えば
a = np.array([[ nan, nan], [ nan, 1], [ 1, 1]])
といった、nanの位置が不定の場合はどうすればよいでしょうか。

[LouiS0616]

どのような結果を期待するかに依ります。
A. [[nan, 1], [1, 1]] ですか？
B. [[1], [1]] ですか？
C. [[1], [1, 1]] ですか？

---
Aの場合... コードの変更は不要です。
Bの場合... np.allの代わりにnp.anyを上手く利用してやります。条件をしっかり考えましょう。
Cの場合... NumPyアレイでは実現不可能です。

[TeRa_130]

ありがとうございます。
Cの場合を想定していたのですが、Numpyでは無理なのですね。
別の方法で試みようと思います。