จัดทำดัชนีทั้งหมด * ยกเว้น * รายการเดียวใน python


113

มีวิธีง่ายๆในการสร้างดัชนีองค์ประกอบทั้งหมดของรายการ (หรืออาร์เรย์หรืออะไรก็ตาม) ยกเว้นดัชนีเฉพาะหรือไม่? เช่น,

  • mylist[3] จะส่งคืนสินค้าในตำแหน่งที่ 3

  • milist[~3] จะส่งคืนรายการทั้งหมดยกเว้น 3

คำตอบ:


112

สำหรับรายการคุณสามารถใช้ list 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]

นี่เป็นเรื่องทั่วไปและสามารถใช้ได้กับการวนซ้ำทั้งหมดรวมถึงอาร์เรย์จำนวนนับ หากคุณแทนที่[]ด้วย(), bจะเป็น iterator แทนของรายการ

หรือคุณสามารถทำได้โดยใช้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])

ซึ่งโดยทั่วไปแล้วจะเร็วกว่าความเข้าใจของรายการที่ระบุไว้ข้างต้นมาก


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

ดูสิ่งนี้ด้วย

อธิบายสัญกรณ์ชิ้นส่วนของ Python


2
คำตอบที่ดีสำหรับรายการ คุณสามารถใช้มันสำหรับอาร์เรย์เกินไป numpy.concatenateแต่คุณจะต้องใช้
Bi Rico

48

วิธีที่ง่ายที่สุดที่ฉันพบคือ:

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

ที่จะผลิตของคุณโดยไม่ต้ององค์ประกอบที่ดัชนีmylistx


ฉันพบว่าสิ่งนี้ลบรายการ x + 1 จริง ๆ แต่ก็ยังมีประโยชน์อยู่ขอบคุณ
Jack 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

+1 สำหรับการปิดบังอาร์เรย์ในกรณีของรายการฉันจะใช้ slice และเชื่อมต่อกันโดยใช้การบีบอัด
Bi Rico

5

ใช้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. ใช้การกำบัง Masking จัดทำโดยฟังก์ชัน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. ใช้ฟังก์ชันitertools.filterfalseจากไลบรารีมาตรฐาน Python

    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

โปรดทราบว่าหากตัวแปรอยู่ในลิสต์ของรายการวิธีการบางอย่างอาจล้มเหลว ตัวอย่างเช่น:

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 ตัวแรกจะถูกลบออกไม่ใช่ตัวที่สอง หวังว่านี่จะยังช่วยได้!

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.