Python获取子列表元素直到N

在本教程中，我们将编写一个程序，该程序返回一个子列表元素，直到列表中的第n个子列表。假设我们有以下列表和5个子列表。

[['Python'，'Java']，['C'，'Pascal']，['Javascript'，'PHP']，['C#'，'C ++']，['React'，'Angular'] ]现在，我们必须从前三个子列表中获取第一个元素。我们可以使元素具有不同的方法。让我们来看一些。

循环

大多数程序员最普遍和最先想到的就是使用循环。我们来看一下使用循环的代码。

示例

# initializing the list and N
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#
C++'], ['React', 'Angular']]
N = 3
# empty list to store final elements from the sublists
desired_elements = []
# iterating over the list till 3rd element
for i in range(N):
   # storing the first element from the sublist
   desired_elements.append(random_list[i][0])
# printing the elements
print(desired_elements)

输出结果

如果运行上面的代码，则将得到以下结果。

['Python', 'C', 'Javascript']

我们可以使用列表推导代替for循环。让我们看看使用列表推导的相同代码。

示例

# initializing the list and N
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#
C++'], ['React', 'Angular']]
N = 3
# getting first element from the sublists
desired_elements = [sublist[0] for sublist in random_list[:N]]
# printing the elements
print(desired_elements)

输出结果

如果运行上面的代码，则将得到以下结果。

['Python', 'C', 'Javascript']

使用内置方法

Python提供了许多内置模块和方法。让我们用它们来解决我们的问题，将使用map，itemgetter和islice方法来实现预期的输出，让我们看一下代码。

示例

# importing the required methods
import operator # for itemgetter
import itertools # for islice
# initializing the list and N
random_list = [['Python', 'Java'], ['C', 'Pascal'], ['Javascript', 'PHP'], ['C#
C++'], ['React', 'Angular']]
N = 3
# getting first element from the sublists
desired_elements = list(map(operator.itemgetter(0), itertools.islice(random_list, N)))
# printing the elements
print(desired_elements)

输出结果

如果运行上面的代码，则将得到以下结果。

['Python', 'C', 'Javascript']

结论

您可以使用任何元素代替第一个元素。我们已将第一个要素用于演示。如果您对本教程有任何疑问，请在评论部分中提及。