python - numpy recarray 最小差异

Question

我有一个 numpy recarray 我想找到记录的第一个元素和最后一个元素的差异最大的记录。

有人可以建议一种方法来做到这一点。

Answer 1

设置：

import numpy
ra = numpy.recarray((10,), 'int,int,int')

查找第一个字段和最后一个字段之间差异的最大值的索引。

idx = numpy.argmax(ra['f0']-ra['f2'])

使用该索引检索记录

print ra[idx]