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

结论

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

以上是 Python获取子列表元素直到N 的全部内容, 来源链接: utcz.com/z/334767.html

回到顶部