[Python]Numpy配列の途中に要素を追加するには？

どうも、ちょげ(@chogetarou)です。

Numpy配列(ndarray)の途中に要素を追加する方法を紹介します。

方法

Numpy配列(ndarray)の途中に要素を追加するには、insert()を使います。

まず、numpyからinsert()を呼び出します。

そして、insert()の第１引数にNumpy配列、第２引数に追加する位置、第３引数に追加する要素を指定します。

#arr=対象のNumpy配列, index=追加する位置, item=追加する要素
result = np.insert(arr, index, item)

上記のinsert()は、Numpy配列(ndarray)の途中に要素を追加した結果を返します。

import numpy as np

nums = np.array([1, 2, 3, 4, 5])

result = np.insert(nums, 2, 10)

print(result)

出力：
[ 1  2 10  3  4  5]