python - 对 NumPy 数组的一行求和

Question

我想对大型 NumPy 数组的某一行求和。我知道函数 array.max() 会给出整个数组的最大值，而 array.max(1) 会给出每一行的最大值数组。但是，我想在某一行(例如第 7 行或第 29 行)中获取最大值。我有一个大数组，因此获取所有行的最大值会给我带来很大的时间损失。

Answer 1

您可以使用索引运算符轻松访问二维数组的一行。行本身是一个数组，是原始数组的一部分的 View ，并暴露了所有数组方法，包括sum() 和max()。因此，您可以像这样轻松获得每行的最大值:

x = arr[7].max()   # Maximum in row 7
y = arr[29].sum()  # Sum of the values in row 29

为了完整起见，您可以对列执行相同的操作:

z = arr[:, 5].sum()  # Sum up all values in column 5.