python - 如何找到给定索引之间的 numpy 数组的最大值？

Question

我已经使用 numpy.genfromtxt 从 csv 加载了一些数据。

这给了我一个数组 (n,3)，其中 n 可以达到 ~500000。

我想在 n=n1 和 n=n2 的值之间的三列中的任何一列中找到最大值。

在 Matlab 中我会很好，但 Python 是一个新的野兽!

(Python 新手，使用 Spyder 和 Anaconda。)

Answer 1

numpy 数组有 max方法。例如

import numpy as np

data = np.arange(30).reshape((10, 3))
n1, n2 = 2, 5
max_value = data[n1:n2, :].max()

max_value 是 14。

如果你想找到每一列的最大值:

max_values = data[n1:n2, :].max(axis=0)

在这种情况下 max_values 是 array([12, 13, 14])