numpy配列を一定の条件で分割する方法

Question

プログラミング、python初心者です！

一次元のndarray配列［1,1,2,3,5,8,13,21,34,55,89］があったとして、この配列を20の倍数を基準として
［1,1,2,3,5,8,13］
[21,34]
[55]
[]
[89]
こんな感じで分割したいです。forループあたり駆使したら実現できるかとは思いますが、より簡潔なコードで実現するにはどのようにかけば良いでしょうか？？　
何かアイデアございましたらご指導お願いいたします。

Answer 1

numpy.digitize() でビン分割するとよいのではないでしょうか。

https://docs.scipy.org/doc/numpy/reference/generated/numpy.digitize.html

import numpy as np

data = np.array([1,1,2,3,5,8,13,21,34,55,89])

cat = np.digitize(data, np.arange(0,120,20))
# [1 1 1 1 1 1 1 2 2 3 5]
ret = [data[cat==i] for i in np.arange(1,6)]
# [array([ 1,  1,  2,  3,  5,  8, 13]),
#  array([21, 34]),
#  array([55]),
#  array([], dtype=int64),
#  array([89])]

Comments

[alicewild]

ありがとうございます！！お陰様でやりたい処理が実現できそうです！