Question

我有一个嵌套列表，如下所示：

A = [('a', 'b', 'c'),
     ('d', 'e', 'f'),
     ('g', 'h', 'i')]

我试图使用代码打印每个列表的第一个元素：

A = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
print A[:][0]

但我得到以下输出：

('a', 'b', 'c')

必需的输出：

('a', 'd', 'g')

如何在Python中获取此输出？

Answer 1

A[:]只需创建整个列表的副本，然后您将获得该副本的元素0。

您需要在此处使用列表解析：

[tup[0] for tup in A]

获取列表，或使用带有生成器表达式的tuple()来获取元组：

tuple(tup[0] for tup in A)

演示：

>>> A = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
>>> [tup[0] for tup in A]
['a', 'd', 'g']
>>> tuple(tup[0] for tup in A)
('a', 'd', 'g')

Answer 2

您可以使用zip(*list_of_lists)转置列表/元组列表，然后选择所需的项目。

>>> a
[('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
>>> b = zip(*a)
>>> b
[('a', 'd', 'g'), ('b', 'e', 'h'), ('c', 'f', 'i')]
>>> b[0]
('a', 'd', 'g')
>>> 
>>> c = zip(*a)[0]
>>> c
('a', 'd', 'g')
>>>

Answer 3

Python列表作为多维数组不能很好地工作。如果您愿意添加额外的依赖项（例如，如果您要进行大量的数组操作），WebApplicationInitializer允许您使用您正在寻找的几乎所有语法

numpy

这会将行输出为np.array（可以像列表一样访问）：

import numpy as np
A = np.array([('a', 'b', 'c'),
              ('d', 'e', 'f'),
              ('g', 'h', 'i')])

将第一行作为元组：

>>> A[:,0]
array(['a', 'd', 'g'])

Answer 4

你也可以这样做：

>>> A = [('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'h', 'i')]
>>> map(lambda t:t[0], A)
['a', 'd', 'g']
>>> tuple(map(lambda t:t[0],A))
('a', 'd', 'g')

Answer 5

您还可以使用pandas获取所需的行为，如下所示：

In [1]: import pandas as pd

In [2]: A = [('a', 'b', 'c'),
     ('d', 'e', 'f'),
     ('g', 'h', 'i')]

In [3]: df = pd.DataFrame(A)

In [4]: df[:][0]
Out[4]:
0    a
1    d
2    g
Name: 0, dtype: object

In [5]: df[:][0].values
Out[5]: array(['a', 'd', 'g'], dtype=object)

如何在Python中索引嵌套列表？

5 个答案: