numpyで配列の配列の配列から配列を抽出したい

Question

pythonのnumpyにおいて、

a=np.array([[[1,2],[3]],[[4,5],[6]],[[7,8],[9]]])
と続くデータ列があります。
この中から、[1,2,4,5,7,8]を1次元配列として取りだす方法を教えてください。
b=a[:,0]
とした場合、b.flatten()で1次元になりません。

よろしくお願いします。

Answer 1

これでどうでしょう？（やってることはリスト内包表記での回答と同じです）

>>> from itertools import chain
>>> a=np.array([[[1,2],[3]],[[4,5],[6]],[[7,8],[9]]])
>>> list(chain.from_iterable(a))
[[1, 2], [3], [4, 5], [6], [7, 8], [9]]
>>> list(chain.from_iterable(a.flatten()))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

Comments

[foo_san]

ご提示のソースを試していないのですが、[1,2,3,4,5,6,7,8,9]を返しているようです。
欲しいのは[1,2,4,5,7,8]です。引き続き回答を募集します。

[shimizukawa]

質問を読み間違えました。 `list(chain.from_iterable(a[:,0]))` としてみてください。

Answer 2

これは、a.の内容がリストであるからだと思います。
以下のようにリスト内包表記で取り出せます。

b = a.flatten()
[ y for x in b for y in x]

Comments

[foo_san]

ご提示のソースを試してみたところ、[1,2,3,4,5,6,7,8,9]が返ってきました。
欲しいのは[1,2,4,5,7,8]です。引き続き回答を募集します。

[foo_san]

b=a[:,0] .flatten()
[ y for x in b for y in x]
でいけました。