如何比较多个列表的第一个元素并在它们相等时附加到新列表？答案

【问题标题】：How do I compare the first elements of multiple lists and append to a new list if they are equal?如何比较多个列表的第一个元素并在它们相等时附加到新列表？
【发布时间】：2017-12-23 20:52:19
【问题描述】：

我有一个数据列表，看起来像这个缩短的数据片段。

[[743, 5043.0, 'N/A', 19.40393776, 7.18, 15.487], [744, 6117.0, 'N/A', 19.22138894, 49.29, 15.341], [745, 5152.0, 'N/A', 16.46983774, 7.94, 15.788], [746, 4856.0, 1.5936507936507938, 9.27358173, 2.47, 15.302], [747, 4537.0, 1.0317880794701988, 6.02930329, 2.85, 15.784], [748, 4989.0, 'N/A', 2.696370652, 1.58, 15.269], [749, 5185.0, 0.8675585284280938, 5.349553819, 2.55, 15.416], [749, 5185.0, 'N/A', 3.94105221, 1.7, 15.416], [749, 5185.0, 'N/A', 8.10904807, 1.3, 15.416]]

现在我有一个列表列表，第一个元素代表 Kepler Object of Interest 编号。我对数据进行了排序，以按第一个元素的顺序显示列表。我的目标是创建一个列表列表，其中所有具有匹配第一个元素的列表都放在一个列表中。例如，有三个带有749 的列表应该放在一个列表中。我正在努力创建一个程序来遍历我的所有列表并将第一个元素与未知数量的其他第一个元素进行比较。最简单的方法是什么？

目标：

[[[743, 5043.0, 'N/A', 19.40393776, 7.18, 15.487]], [[744, 6117.0, 'N/A', 19.22138894, 49.29, 15.341]], [[745, 5152.0, 'N/A', 16.46983774, 7.94, 15.788]], [[746, 4856.0, 1.5936507936507938, 9.27358173, 2.47, 15.302]], [[747, 4537.0, 1.0317880794701988, 6.02930329, 2.85, 15.784]], [[748, 4989.0, 'N/A', 2.696370652, 1.58, 15.269]], [[749, 5185.0, 0.8675585284280938, 5.349553819, 2.55, 15.416], [749, 5185.0, 'N/A', 3.94105221, 1.7, 15.416], [749, 5185.0, 'N/A', 8.10904807, 1.3, 15.416]]]

【问题讨论】：

标签： list python-3.x indexing compare elements

【解决方案1】：

我认为将列表列表转换为 pandas DataFrame 是一个不错的方法：

import pandas as pd
x = pd.DataFrame([[743, 5043.0, 'N/A', 19.40393776, 7.18, 15.487], [744, 6117.0, 'N/A', 19.22138894, 49.29, 15.341], [745, 5152.0, 'N/A', 16.46983774, 7.94, 15.788], [746, 4856.0, 1.5936507936507938, 9.27358173, 2.47, 15.302], [747, 4537.0, 1.0317880794701988, 6.02930329, 2.85, 15.784], [748, 4989.0, 'N/A', 2.696370652, 1.58, 15.269], [749, 5185.0, 0.8675585284280938, 5.349553819, 2.55, 15.416], [749, 5185.0, 'N/A', 3.94105221, 1.7, 15.416], [749, 5185.0, 'N/A', 8.10904807, 1.3, 15.416]])
y = x.groupby(0).apply(lambda z: list(z.values))
Goal = [[list(z) for z in y[idx]] for idx in y.index]

我相信这是按照您描述的方式进行格式化和分组的。

【讨论】：

谢谢你，这正是我所需要的。

【解决方案2】：

如果你想在 Python 中使用内置功能，你可以使用字典。

from collections import OrderedDict

l = [[743, 5043.0, 'N/A', 19.40393776, 7.18, 15.487], [744, 6117.0, 'N/A', 19.22138894, 49.29, 15.341], [745, 5152.0, 'N/A', 16.46983774, 7.94, 15.788], [746, 4856.0, 1.5936507936507938, 9.27358173, 2.47, 15.302], [747, 4537.0, 1.0317880794701988, 6.02930329, 2.85, 15.784], [748, 4989.0, 'N/A', 2.696370652, 1.58, 15.269], [749, 5185.0, 0.8675585284280938, 5.349553819, 2.55, 15.416], [749, 5185.0, 'N/A', 3.94105221, 1.7, 15.416], [749, 5185.0, 'N/A', 8.10904807, 1.3, 15.416]]
d = OrderedDict()
for inner in l:
    if inner[0] in d:
        d[inner[0]].append(inner)
    else:
        d[inner[0]] = [inner]

list(d.values())

我使用了 OrderedDict，所以它保留了原始项目的顺序。

【讨论】：