我有一个结构化数组的对象列表,例如这样的:
a = np.array([('Rex', 9, 81.0), ('Fido', 3, 27.0)], dtype=[('name', 'U10'), ('age', 'i4'), ('weight', 'f4')])
b = np.array([('Dog3', 9, 81.0), ('Dog4', 3, 27.0)], dtype=[('name', 'U10'), ('age', 'i4'), ('weight', 'f4')])
c = np.array([('Dog5', 9, 81.0), ('Dog6', 3, 27.0)], dtype=[('name', 'U10'), ('age', 'i4'), ('weight', 'f4')])
lst = [a, b, c]
现在我需要这个列表本身就是一个 numpy 数组,因为我需要numpy.where()
在它上面使用,否则这不起作用。
lst = np.array(lst)
那么我做这样的事情:
ID = np.where(lst == c)
lst[ID] = 0 or rather lst[ID] = None
但不是我想得到的,即
>>>lst
array([a, b, 0/None], dtype=...)
我要么得到这个:
>>>lst
array([a, b, [('0', 0, 0.), ('0', 0, 0.)]], dtype=...)
或者它根本不起作用:
TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'
我怎样才能做到这一点?我必须转换lst
回列表拳头吗?
让我们创建一个更简单的数组,并尝试设置一些值:
In [14]: dt = np.dtype([('foo','U10'),('bar',int)])
In [16]: arr = np.zeros(3, dtype=dt)
In [17]: arr
Out[17]: array([('', 0), ('', 0), ('', 0)], dtype=[('foo', '<U10'), ('bar', '<i8')])
In [18]: arr[1]
Out[18]: ('', 0)
In [19]: arr[1] = 12
In [20]: arr
Out[20]:
array([('', 0), ('12', 12), ('', 0)],
dtype=[('foo', '<U10'), ('bar', '<i8')])
注意字符串 '12' 和整数 12 的混合。
设置一个元组,每个字段一个值:
In [21]: arr[2] = ('dog',23)
使用 a 设置None
失败,因为它无法None
按照第二个字段的要求转换为整数。
In [22]: arr[0] = None
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-22-2f7a4c897706> in <module>
----> 1 arr[0] = None
TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'
In [23]: arr
Out[23]:
array([('None', 0), ('12', 12), ('dog', 23)],
dtype=[('foo', '<U10'), ('bar', '<i8')])
实际上它确实设法设置了字符串字段。
arr
是一个数组,其中每个元素必须具有相同的dtype
.
带有object
dtype的数组的行为更像是一个列表。虽然偶尔有用,但不应用作列表的替代品。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句