どうも、ちょげ(@chogetarou)です。
配列(array)の指定したインデックスに要素を追加する方法を紹介します。
リストの場合

リストの指定したインデックスに要素を追加するには、insertメソッドを使います。
まず、リストからinsertメソッドを呼び出します。
そして、insertメソッドの第1引数にインデックス、第2引数に追加する要素を指定します。
#ls=対象のリスト, index=インデックス, item=追加する要素
ls.insert(index, item)
insertメソッドは、呼び出したリストの第1引数の位置に第2引数の要素を追加します。
使用例
numbers = [1, 2, 4, 5]
numbers.insert(2, 3)
numbers.insert(1, 10)
print(numbers) #[1, 10, 2, 3, 4, 5]
Numpy配列の場合

Numpy配列(ndarray)の途中に要素を追加するには、insert()を使います。
まず、numpyからinsert()を呼び出します。
そして、insert()の第1引数にNumpy配列、第2引数に追加する位置、第3引数に追加する要素を指定します。
#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]
まとめ
リスト(List)の指定したインデックスに要素を追加するには、insertメソッドを使います。
#ls=対象のリスト, index=インデックス, item=追加する要素
ls.insert(index, item)
Numpy配列(ndarray)の指定したインデックスに要素を追加するには、numpy.insert()を使います。
#arr=対象のNumpy配列, index=追加する位置, item=追加する要素
result = np.insert(arr, index, item)
コメント