python - 将多个列表连接成一个数据结构

Question

使用 Python，我有两个长度相同的列表，其中的每个数据都以相同的顺序相互关联。因此，我做了一个列表列表如下:

a1=[0, 1, 1, 2]
a2=[o, x, o, o]
-> A=[(0,o), (1,x), (1,o), (2,o)]

后来，我发现了另一个我想添加的列表:

a3=[0,0,1,1]

虽然我可以通过调用前面的列表 a1 和 a2 来连接列表，但我想知道是否有任何其他简单的方法可以仅使用 A 和 a2 来连接它，因为如果已经连接了很多列表，这似乎很困难这边走;我想要一种可以在将来轻松扩展的方法。

此外，如果在这种情况下制作列表列表不是最佳方式，我想知道我应该使用哪些其他结构。

Answer 1

您可以使用 Pandas 轻松做到这一点。由于所有列表的长度都相同，因此您可以将每个列表视为 Pandas DataFrame 中的一列。例如:

import pandas as pd

a1 = [0, 1, 1, 2]
a2 = [o, x, o, o]
A = pd.DataFrame({'a1': a1, 'a2': a2})

现在你有这样的东西:

  a1 a2
0  0  o
1  1  x
2  1  o
3  2  o

您可以向其中添加列:

a3 = [0,0,1,1]
A['a3'] = a3

第一个未命名的列是“索引”，因此您可以像这样使用它:

>>> print(A.loc[0])
0 o 0

您还可以访问特定的列:

>>> print(A['a3'])
0 0 1 1

此解决方案比列表列表的效率高得多，并且会随着数据的增长提供更快的性能。例如，您可以使用 A['a1'].sum() 非常快速地对整个列求和，或者使用 A['a1'] + A['a3']< 添加列(给出 0 1 2 3)。