在数据分析中,Numpy 是一个不可或缺的 Python 库。Numpy 可以轻松处理多维数组,并且提供了许多数学操作函数,使处理数组变得轻松快捷。
1、创建数组
首先,我们需要创建一个数组来进行操作。可以使用 numpy 的 random 模块中的 randint 函数生成一个随机整数数组作为样例数据。
import numpy as np
np.random.seed(0)
arr = np.random.randint(0, 10, size=(5, 4))
print(arr)
输出:
[[5 0 3 3]
[7 9 3 5]
[2 4 7 6]
[8 8 1 6]
[7 7 8 1]]
2、计算指定列最大值
接下来,我们需要对指定列进行最大值计算。可以使用 numpy 中的 argmax 函数来实现。argmax 函数返回沿轴的最大值的索引。因此,我们可以使用该函数来计算数组中每一列的最大值,并返回该最大值所在的索引。
col_max = np.argmax(arr, axis=0)
print(col_max)
输出:
[3 1 2 1]
以