python - 在 numpy.where 中访问对象方法？

Question

我有一个 numpy 数组中的数据(从 .csv 文件中读取)。 np.genfromtxt 的相关摘录是:

dtype = [("Category", "|S10"),
         ("Status", "|S11"),
         ("Date_start", object),
         ("Date_stop", object)],
names=True,
converters={2:lambda d:datetime.strptime(d, "%d/%m/%y"),
            3:lambda d:datetime.strptime(d, "%d/%m/%y")}
)

一切正常，只有一个异常(exception)——访问日期时间对象的元素。以下两行代码返回的正是我所期望的:

print inp['Date_start'][1].month #returns 7
print np.where(inp['Category'] == '"R5"') #returns an array of matching indices

但下面的代码行抛出一个 AttributeError: 'numpy.ndarray' object has no attribute 'month'

print np.where(inp['Date_start'].month == 7)

这意味着我无法根据事情发生的月份返回结果，而我需要这样做。

有什么方法可以从 np.where 获得我想要的行为吗？

Answer 1

您可以定义一个矢量化属性 getter :

def func(a):
    return a.month

vfunc = np.vectorize(func)

然后使用:

np.where(vfunc(inp['Date_start']) == 7)