Pythonでリストの中央からK個の要素を抽出する方法

はじめに

リストから中央付近のK個の要素を取り出したい場合、Pythonでは「//」演算子（切り捨て除算）とリストスライシングを組み合わせることで、シンプルかつ効率的に実現できます。

本記事では、具体的なサンプルコードとともに、その仕組みをわかりやすく解説します。

サンプルコード

my_list = [34, 56, 12, 67, 88, 99, 0, 1, 21, 11]

print("The list is : ")
print(my_list)

K = 5
print("The value of K is ")
print(K)

beg_indx = (len(my_list) // 2) - (K // 2)
end_indx = (len(my_list) // 2) + (K // 2)

my_result = my_list[beg_indx: end_indx + 1]

print("The result is : " )
print(my_result)

実行結果

The list is :
[34, 56, 12, 67, 88, 99, 0, 1, 21, 11]
The value of K is
5
The result is :
[67, 88, 99, 0, 1]

処理の流れと解説

• まず対象となるリストを定義し、コンソールに表示します。

• 取得したい要素数Kを定義し、こちらもコンソールに表示します。

• len()関数でリストの長さを取得し、「//」演算子を使って2で割ることで、リストの中央位置を求めます。

• 中央位置から「K // 2」を引いた値を開始インデックス（beg_indx）として変数に代入します。

• 同様に、中央位置に「K // 2」を加えた値を終了インデックス（end_indx）として別の変数に代入します。

• リストスライシングを行い、開始インデックスから終了インデックスまでの範囲の要素にアクセスします。ここでend_indxに1を加えているのは、スライシングの終端が排他的（その位置自体は含まれない）であるためです。

• 抽出した結果を変数my_resultに代入します。

• 最後に、その結果をコンソールへ出力します。

補足ポイント

この手法では、リストの要素数が偶数の場合でも「//」による切り捨て除算のおかげで、常に中央寄りの位置からK個の要素を安定して取得できます。また、Kがリストの長さより大きい場合でもエラーにはなりませんが、取得できる要素数はリストの範囲内に制限される点に注意してください。