Python| K元素切片

我们经常遇到需要提取list的初始K个元素的情况。当我们需要优化内存时，就会出现这个特殊问题。当有时我们需要获取所有相似大小的列表时，这在日常编程中有它的应用。让我们讨论可以执行此任务的某些方式。

方法 #1：使用len() + 列表切片
列表切片可以执行这个特定的任务，我们只需将前 K 个元素切片到列表中，从而删除剩余的元素，释放内存。

# Python code to demonstrate 
# K elements Slicing
# using len() + list slicing
  
# initializing list 
test_list = [1, 4, 6, 3, 5, 8]
  
# printing original list
print ("The original list is : " + str(test_list))
  
# initializing K 
K = 4
  
# using len() + list slicing
# K elements Slicing
res = test_list[: K]
  
# printing result 
print ("The K sliced List : " + str(res))

输出 ：

The original list is : [1, 4, 6, 3, 5, 8]
The K sliced List : [1, 4, 6, 3]

方法#2：使用负列表切片
我们可以使用否定列表切片来执行此特定任务，在该切片中，我们开始从列表的最后一个索引中删除元素，从而从最后一个索引中删除所有 N - K 个元素。如果要求删除 0 个元素，我们将删除 None。

# Python code to demonstrate 
# K elements Slicing
# using negative list slicing
  
# initializing list 
test_list = [1, 4, 6, 3, 5, 8]
  
# printing original list
print ("The original list is : " + str(test_list))
  
# initializing K 
K = 4
  
# using negative list slicing
# K elements Slicing
res = test_list[ : -(len(test_list) - K)]
  
# printing result 
print ("The K sliced List : " + str(res))

输出 ：

The original list is : [1, 4, 6, 3, 5, 8]
The K sliced List : [1, 4, 6, 3]