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


假设我现在有一个结构数组,其中包含有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)



阅读更多
个人分类: python
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

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

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭