查找列中的最大值并返回行号

Question

我想找出哪个行号包含最大值。

我尝试过这样的事情：

specimen = loadtxt('dataFile.dat')
stress = specimen[:,3]*(1+(specimen[:,2]/100))
strain = log(1+ specimen[:,2]/100)*100

m = max(stress) #<-- at which row??
print m
print row

Answer 1

您可以使用NumPy的argmax()功能或方法。

stress.argmax()

将返回数组中最大值的索引。您可以选择指定和轴以获得沿所选轴的所有最大索引。

Answer 2

maxima = [max(row) for row in stress]  # Find the max values in each row
m = max(maxima)                        # Find the absolute max value in the array
for k, rowmax in enumerate(maxima):
    if rowmax == m:                    # Find which row contains the absolute max value
        row = k
        break

print m
print row