索引所有*除了* python中的一个项目

时间:2013-10-10 03:39:18

标签: python list numpy indexing

是否有一种简单的方法可以为特定索引的列表(或数组,或其他)之外的所有元素编制索引?例如,

  • mylist[3]将返回位置3

  • 中的项目
  • milist[~3]将返回除3

  • 之外的整个列表

10 个答案:

答案 0 :(得分:75)

对于列表,您可以使用列表comp。例如,要使b成为a的副本,而不包含第3个元素:

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 中,您可以使用布尔索引执行此操作:

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])

一般来说,它比上面列出的列表理解要快得多。

答案 1 :(得分:45)

>>> l = range(1,10)
>>> l
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> l[:2] 
[1, 2]
>>> l[3:]
[4, 5, 6, 7, 8, 9]
>>> l[:2] + l[3:]
[1, 2, 4, 5, 6, 7, 8, 9]
>>> 

另见

Explain Python's slice notation

答案 2 :(得分:34)

我找到的最简单的方法是:

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

将产生mylist而没有索引为x的元素。

答案 3 :(得分:22)

如果你正在使用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]

答案 4 :(得分: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]
    

答案 5 :(得分:1)

使用np.delete!它实际上并不会删除任何内容

示例:

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

In [6]: a                                                                       
Out[6]: 
array([[1, 4],
       [5, 7],
       [3, 1]])

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

In [8]: ind                                                                     
Out[8]: array([0, 1])

In [9]: a[ind]                                                                  
Out[9]: 
array([[1, 4],
       [5, 7]])

In [12]: np.delete(a, ind, axis=0)                                              
Out[12]: array([[3, 1]])

In [13]: a                                                                      
Out[13]: 
array([[1, 4],
       [5, 7],
       [3, 1]])

答案 6 :(得分:0)

如果您事先不知道索引,那么这是一个可行的功能

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

答案 7 :(得分: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

答案 8 :(得分:0)

如果要剪掉最后一个或第一个,请执行以下操作:

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

如果将1更改为2,则将删除前2个字符,而不是第二个。 希望这对您有所帮助!

答案 9 :(得分:0)

arr=[1,3,5,7,9]
for i in range(len(arr)):
    arsum = arr[:i] + arr[i + 1:]