Numpyの連番を抜き出すアルゴリズム

ソート済みで重複のない配列中に現れる連番（隣り合うインデックスの差が１）を見つける方法を思案しています．コードの可読性よりも計算速度を重視します．

例えば，
L = [1,2,5,9,10,11,13,15,16,19]
というnumpy配列があるとき（ソート済みで重複なし）

[[1,2],
[5],
[9,10,11],
[13],
[15,16],
[19]]

というようなリストを作りたいんです．どのようにすると良いでしょうか？
配列Lのlengthは5000程度です．

具体的なコードがなくても，アルゴリズムだけでも指南して頂ければ助かります．
（もちろんコードもくれると嬉しいです）

とりあえず思いつくのはこんな感じです．

python
1x = np.array([1,2,5,9,10,11,13,15,16,19])
2
3new = []
4tmp = []
5for i in range(len(x)):
6    if x[i] == x[i-1] + 1:
7        tmp.append(x[i])
8    else:
9        new.append(tmp)
10        tmp = []
11        tmp.append(x[i])
12print(new)

行動規範の内容に同意します

回答1件

ベストアンサー

質問のタイトルにnumpyとありますが、コードがリストでしたのでリストで回答します。

Python
1result = []
2tmp = [x[0]]
3for i in range(len(x)-1):
4    if x[i+1] - x[i] == 1:
5        tmp.append(x[i+1])
6    else:
7        if len(tmp) > 0:
8            result.append(tmp)
9        tmp = []
10        tmp.append(x[i+1])
11
12print(result)
13# [[1, 2], [5], [9, 10, 11], [13], [15, 16]]