关闭

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

784人阅读 评论(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)



0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:76004次
    • 积分:1376
    • 等级:
    • 排名:千里之外
    • 原创:60篇
    • 转载:3篇
    • 译文:1篇
    • 评论:16条
    文章分类
    最新评论