python - 从具有数据类型的 NumPy 数组中切片一列？

Question

我构造了一个NumPy数组并用数据填充它：

data = np.zeros(ndata,dtype=[('datetime',np.dtype(np.datetime64)),\
    ('Value','<f8'),('Weight','<f8')])

但是，现在我想做一些事情，比如只返回datetime“列”或只返回值。但这不是一个2D NumPy数组。这是1D一些 custom 的数组dtype，所以我不能这样做：

data[:,0]

我认为这是微不足道的，或者我从根本上对如何使用NumPy数组有错误的想法。

Answer 1

numpy 数据类型可以像字典一样被访问，所以你可以这样做：

data[0]['datetime']

获取第零行的日期时间字段或

data[:]['datetime']

获取所有行的日期时间字段数组。