Pythonで1つの項目を除くすべてのインデックス


112

特定のインデックスを除いて、リスト(または配列など)のすべての要素にインデックスを付ける簡単な方法はありますか?例えば、

  • mylist[3] アイテムを位置3に戻します

  • milist[~3] 3を除くリスト全体を返します

回答:


111

以下のためにリスト、リストカンプを使用することができます。たとえば、3番目の要素なしでのbコピーを作成するにはa

a = range(10)[::-1]                       # [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
b = [x for i,x in enumerate(a) if i!=3]   # [9, 8, 7, 5, 4, 3, 2, 1, 0]

これは非常に一般的で、numpy配列を含むすべての反復可能オブジェクトで使用できます。で置き換える[]()bリストではなくイテレータになります。

または、次のようにインプレースで実行できますpop

a = range(10)[::-1]     # a = [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
a.pop(3)                # a = [9, 8, 7, 5, 4, 3, 2, 1, 0]

numpyのあなたは、boolean型インデックスでこれを行うことができます:

a = np.arange(9, -1, -1)     # a = array([9, 8, 7, 6, 5, 4, 3, 2, 1, 0])
b = a[np.arange(len(a))!=3]  # b = array([9, 8, 7, 5, 4, 3, 2, 1, 0])

これは一般に、上記のリスト内包表記よりもはるかに高速です。



48

私が見つけた最も簡単な方法は:

mylist[:x]+mylist[x+1:]

mylistインデックスの要素なしであなたを生成しますx


私はこれが実際にアイテムx + 1を削除することを発見しましたが、それでもなお有用です、ありがとう
ジャックTC

24

numpyを使用している場合は、最も近い、マスクを使用していると考えることができます

>>> import numpy as np
>>> arr = np.arange(1,10)
>>> mask = np.ones(arr.shape,dtype=bool)
>>> mask[5]=0
>>> arr[mask]
array([1, 2, 3, 4, 5, 7, 8, 9])

を使用itertoolsせずに同様のことが実現できますnumpy

>>> from itertools import compress
>>> arr = range(1,10)
>>> mask = [1]*len(arr)
>>> mask[5]=0
>>> list(compress(arr,mask))
[1, 2, 3, 4, 5, 7, 8, 9]

2
np.arange(len(arr)) != 3はマスクのようなものを使用するかもしれませんarr[~(np.arange(len(arr)) == 3)]
DSM

@DSM:これを回答として投稿してください:-)。いずれにせよ、私はNumpyに精通していません。
Abhijit 2013年

配列をマスクするための+1。リストの場合、スライスを使用し、compressを使用して連結します。
Bi Rico

4

使用してくださいnp.delete!実際には何も削除されません

例:

import numpy as np
a = np.array([[1,4],[5,7],[3,1]])                                       

# a: array([[1, 4],
#           [5, 7],
#           [3, 1]])

ind = np.array([0,1])                                                   

# ind: array([0, 1])

# a[ind]: array([[1, 4],
#                [5, 7]])

all_except_index = np.delete(a, ind, axis=0)                                              
# all_except_index: array([[3, 1]])

# a: (still the same): array([[1, 4],
#                             [5, 7],
#                             [3, 1]])

2

それを行うための機能的な(不変の)方法を提供します。

  1. それを行うための標準的で簡単な方法は、スライスを使用することです。

    index_to_remove = 3
    data = [*range(5)]
    new_data = data[:index_to_remove] + data[index_to_remove + 1:]
    
    print(f"data: {data}, new_data: {new_data}")

    出力:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  2. リスト内包表記を使用:

    data = [*range(5)]
    new_data = [v for i, v in enumerate(data) if i != index_to_remove]
    
    print(f"data: {data}, new_data: {new_data}") 

    出力:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  3. フィルター機能を使用:

    index_to_remove = 3
    data = [*range(5)]
    new_data = [*filter(lambda i: i != index_to_remove, data)]

    出力:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  4. マスキングを使用します。マスキングは、標準ライブラリのitertools.compress関数によって提供されます。

    from itertools import compress
    
    index_to_remove = 3
    data = [*range(5)]
    mask = [1] * len(data)
    mask[index_to_remove] = 0
    new_data = [*compress(data, mask)]
    
    print(f"data: {data}, mask: {mask}, new_data: {new_data}")

    出力:

    data: [0, 1, 2, 3, 4], mask: [1, 1, 1, 0, 1], new_data: [0, 1, 2, 4]
  5. Python標準ライブラリのitertools.filterfalse関数を使用する

    from itertools import filterfalse
    
    index_to_remove = 3
    data = [*range(5)]
    new_data = [*filterfalse(lambda i: i == index_to_remove, data)]
    
    print(f"data: {data}, new_data: {new_data}")

    出力:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]

0

事前にインデックスがわからない場合は、ここで機能する関数です

def reverse_index(l, index):
    try:
        l.pop(index)
        return l
    except IndexError:
        return False

0

変数がリストのリストである場合、いくつかのアプローチは失敗することに注意してください。例えば:

v1 = [[range(3)] for x in range(4)]
v2 = v1[:3]+v1[4:] # this fails
v2

一般的なケースでは、

removed_index = 1
v1 = [[range(3)] for x in range(4)]
v2 = [x for i,x in enumerate(v1) if x!=removed_index]
v2

0

最後または最初を切り取りたい場合は、次のようにします。

list = ["This", "is", "a", "list"]
listnolast = list[:-1]
listnofirst = list[1:]

1から2に変更すると、最初の2文字は削除され、2番目の文字は削除されません。これがまだ役立つことを願っています!

弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.