Question

我需要在具有两列元组/序列的数据框中找到元组/序列的精确匹配，如下所示：

Tuple = (10, 0, 0)

pandas.Dataframe =             From                    To
                    2      (10.0, 0.0, 0.0)      (20.0, 0.0, 0.0)
                    3     (10.0, -0.0, 0.0)    (18.66, -5.0, 0.0)
                    4     (10.0, -0.0, 0.0)    (15.0, -8.66, 0.0)
                    5    (15.0, -8.66, 0.0)   (20.0, -17.32, 0.0)
                    6   (20.0, -17.32, 0.0)  (11.34, -12.32, 0.0)
                    7    (15.0, -8.66, 0.0)    (25.0, -8.66, 0.0)
                    8  (11.34, -12.32, 0.0)   (1.34, -12.32, 0.0)
                    9   (1.34, -12.32, 0.0)   (10.0, -17.32, 0.0)

此处应返回索引2、3、4

谢谢

PS。我说元组/系列是因为我不知道哪一个效率最高。我可以使用或

Answer 1

使用此：

print(DataFrame[(DataFrame['From'] == Tuple) | (DataFrame['To'] == Tuple)].index.values)

在数据框中搜索确切的元组/列表/序列

1 个答案: