numpy中结构数组的多个字段进行组合选择

最新推荐文章于 2023-09-16 17:29:55 发布

tumin999

最新推荐文章于 2023-09-16 17:29:55 发布

阅读量1.4k

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/tumin999/article/details/17092107

版权

python 专栏收录该内容

32 篇文章 0 订阅

订阅专栏

假设我现在有一个结构数组，其中包含有nDate nTime两项。如果用np.argwhere()直接判断指定的日期时间，该怎么写呢？

import numpy as np

dtype = np.dtype({'names':['nDate', 'nTime', 'data'],'formats':['i','i', 'i']})
data = np.zeros(10, dtype=dtype)
data[1]['nDate'] = 20130104
data[1]['nTime'] = 102345

idxs = np.argwhere (data[['nDate', 'nTime']] == np.array((20130104, 102345), dtype=dtype.descr[0:2]))
print (idxs)