在数组列表中查找数组存在的更好方法

Question

我是 Python 的新手，正在尝试不同的方法来优化和简化我的代码。

我有一个数组列表（必须是这种格式）最初是空的，我需要用数组更新它，确保不添加重复的条目。

现在我正在按照以下方式进行操作，这是我尝试过的唯一有效的方法：

if len(where(((array(self.pop_next)-(self.pop[self.top_indv_indx[i]]))==0).sum(1)==len((self.pop[self.top_indv_indx[i]])))[0])<=0):
     self.pop_next.append(self.pop[self.top_indv_indx[i]])

self.pop_next 是我的数组列表，self.pop[self.top_indv_indx[i]] 是要添加的数组。

我知道这个 Unpythonic 并且猜想有更好的简单方法来做同样的事情。 请帮忙

Answer 1

您可能想尝试使用numpy.all(array1 == array2) 作为单个数组比较的条件。

编辑中的扩展：

要遍历列表，您可以使用以下内容：

if all((numpy.all(array_to_add != a) for a in array_list)):
    array_list.append(array_to_add)

这会将array_to_add 与array_list 的所有元素按值进行比较。请注意，这里的all 是__builtin__.all，与numpy.all 形成对比。如果您之前做过from numpy import *，这将不起作用。请改用import numpy，并按照上面的示例按全名调用函数。

如果可以按对象进行比较（即两个数组只有在内存中的对象完全相同时才相同），请使用以下更简单的变体：

if array_to_add is not in array_list:
    array_list.append(array_to_add)

Answer 2

编辑：我从您的评论中看到您正在使用 numpy 数组。我从来没有使用过 numpy，所以我不知道它们是如何处理集合的。

一种选择是使用set。 Sets are like lists 但它们是无序的，并且只允许每个项目添加一次：

>>> s = set()
>>> s.add(1)
>>> s.add(2)
>>> s.add(2)
>>> s.add(2)
>>> s
set([1, 2])

但是，如果您尝试将 list 添加到集合中，则会遇到问题：

>>> s.add(['my','list'])
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unhashable type: 'list'

要添加到 set 的项目 must be hashable 和 list 不能是可散列的，因为它不能具有不变的散列值，因为它可以随时通过添加或删除值来修改。

如果您不需要要检查的列表是可变的，您可以将它们转换为固定的、可散列且易于设置的元组：

>>> mylist = ['my','list']
>>> s = set()
>>> s.add(tuple(mylist))
>>> s.add(tuple(mylist))
>>> s
set([('my', 'list')])