多次元配列の一次元化

Question

多次元配列を一次元化したいです。

[[1,2,4],[2,3]]

#[1,2,4,2,3]

ここでNumpyのravel関数を使いました。

np.ravel([[1,2,4],[2,3]])

#array([list([1, 2, 4]), list([2, 3])], dtype=object)

このように出力結果は一次元配列ではありませんでした。
Numpyのflatten関数を使っても同じ結果になってしまいました。

組み込み関数であるsumを使ってみました。

sum([[1,2,4],[2,3]],[])

#[1,2,4,2,3]

出力結果は望んでいたものでした。
しかし、とても大きな多次元配列を用いるためsumを使った結果フリーズしてしまいました。

そのためNumpyのsum関数はまだ試せていない状態ですが、おそらく望んでいるものが出力されるでしょう。
しかしまたフリーズしてしまったら恐ろしいので他のより効率的な関数はあるでしょうか、それともNumpyのsum関数は十分高速でしょうか。

Answer 1

itertools.chain.from_iterableを使って、listに変換されては。効率的には一番良いやり方のはずです。

from itertools import chain

list(chain.from_iterable([[1,2,4],[2,3]]))

https://docs.python.org/ja/3/library/itertools.html#itertools.chain.from_iterable

Comments

[hoshi1996]

ありがとうございます。
試してみます

Answer 2

numpy形式のものをnp.sumすれば通常よりは高速だと思いますが、フリーズの理由がメモリっぽいですね...。イテレータを使用してメモリを節約することをお勧めします。

#a = np.array([[1,2,4],[2,3]])
#np.sum(a)

# example
a = iter([[1,2,4],[2,3],[1,2,3],[3,4,5,6]])
res=[]
for i in a:
    #print(flatten([i,next(a)]))
    res = flatten([res,i])
    return res

Comments

[hoshi1996]

ありがとうございます。
試してみます