2次元配列から0以外の要素を抽出したい。

Question

前提・実現したいこと

Pythonで2次元配列から0以外の要素を抽出したいです。
但し,列ごとに抽出したいです。例えば、

[[0.    0.    0.    0.    0.    0.12  0.    0.    0.    0.   ]
 [0.    0.    0.    0.    0.    0.    0.    0.    0.    0.35 ]
 [0.43  0.434 0.432 0.    0.    0.    0.    0.    0.    0.   ]
 [0.    0.    0.    0.74  0.62  0.    0.    0.    0.76  0.   ]
 [0.    0.    0.    0.    0.    0.    0.87  0.99  0.    0.   ]]

この様な配列の場合、

 [[0.12]
  [0.35]
  [0.43  0.434 0.432]
  [0.74  0.62  0.76]
  [0.87  0.99]]

この様にして配列として取得したいです。

試したこと

以下の様に行うと

import numpy as np

scale = 5
list1 = [0.43, 0.434, 0.432, 0.74, 0.62,0.12,0.87,0.99,0.76,0.35]
list2 = np.zeros((scale,len(list1)))

for i, v in enumerate(list1):
  list2[int(abs(v)*scale),i] = v

print(list2) #list2に今回分類したい配列が入っています。

list3=(list2[list2>0])
print(list3)

[0.12  0.35  0.43  0.434 0.432 0.74  0.62  0.76  0.87  0.99 ]

1次元で全部纏められてしまいダメでした。

補足情報（FW/ツールのバージョンなど）

Python　3.9.4

Answer 1

• この様にして配列として取得したいです。

numpy.ndarrayではそういう配列は存在できません。

リストとして取得するなら以下です。

>>> list3 = [[e for e in row if e!=0] for row in list2.tolist()]
>>> print(list3)
[[0.12], [0.35], [0.43, 0.434, 0.432], [0.74, 0.62, 0.76], [0.87, 0.99]]

後ろにNaNを詰めた形のndarrayにするなら以下を追加すればできます。

>>> import pandas as pd
>>> print(pd.DataFrame(list3).values)
[[0.12    nan   nan]
 [0.35    nan   nan]
 [0.43  0.434 0.432]
 [0.74  0.62  0.76 ]
 [0.87  0.99    nan]]

Comments

[ntsk2543]

迅速な回答、ありがとうございました。

Answer 2

numpy.array のリストで良ければ。。。

>>> from pprint import pprint
>>> pprint([x[~np.isnan(x)] for x in np.where(list2>0, list2, np.nan)])

[array([0.12]),
 array([0.35]),
 array([0.43 , 0.434, 0.432]),
 array([0.74, 0.62, 0.76]),
 array([0.87, 0.99])]