如何检查numpy数组是否包含空列表

Question

这是数据的示例代码

import numpy as np
myList1 = np.array([1,1,1,[1],1,1,[1],1,1])
myList2 = np.array([1,1,1,[],1,1,[],1,1])

要查看 myList1 中的元素是否等于 [1]，我可以这样做：

myList1 == [1]

但是对于 myList2，要查看 myList2 中的元素是否等于 [] 我不能这样做：

myList2 == []

我必须做：

[x == [] for x in myList2]

是否有另一种方法可以在列表中查找也可以处理空列表的元素？ 我可以使用 numpy 或 python 中的其他一些函数吗？

Answer 1

混合了数字和列表（空或非空）的数组是object dtype 。 这实际上是一个list ； 快速编译的numpy数学不再有效。 列表np.frompyfunc的唯一实用替代方案是np.frompyfunc 。

编写一个可以区分数字和列表以及列表长度的小函数，并将其应用于数组。 如果它为空列表返回 True，则np.where将标识位置

In [41]: myList1 = np.array([1,1,1,[1],1,1,[1],1,1]) 
    ...: myList2 = np.array([1,1,1,[],1,1,[],1,1])                              

开发一个函数，对于空列表返回 True，否则返回 False：

In [42]: len(1)                                                                 
...
TypeError: object of type 'int' has no len()
In [43]: len([])                                                                
Out[43]: 0

In [44]: def foo(item): 
    ...:     try: 
    ...:         return len(item)==0 
    ...:     except TypeError: 
    ...:         pass 
    ...:     return False 
    ...:                                                                        
In [45]: foo([])                                                                
Out[45]: True
In [46]: foo([1])                                                               
Out[46]: False
In [47]: foo(1)                                                                 
Out[47]: False

将其应用于数组：

In [48]: f=np.frompyfunc(foo,1,1)                                               
In [49]: f(myList1)                                                             
Out[49]: 
array([False, False, False, False, False, False, False, False, False],
      dtype=object)
In [50]: np.where(f(myList1))                                                   
Out[50]: (array([], dtype=int64),)
In [51]: np.where(f(myList2))                                                   
Out[51]: (array([3, 6]),)