以下是一个示例代码,实现了按照三列进行分区并找到每个分区的最大值。
import numpy as np
# 创建一个3列10行的随机数组
arr = np.random.randint(0, 100, size=(10, 3))
# 按照三列进行分区
partitions = np.hsplit(arr, 3)
# 找到每个分区的最大值
max_values = [np.max(partition) for partition in partitions]
print("分区后的数组:")
print(arr)
print("每个分区的最大值:")
print(max_values)
这段代码使用了NumPy库来处理数组。首先,我们创建了一个随机数组arr
,它有10行和3列。然后,使用np.hsplit()
函数将数组按照列进行分区,得到一个包含分区后的数组的列表partitions
。接下来,我们使用循环遍历每个分区,并使用np.max()
函数找到每个分区的最大值,并将它们保存在max_values
列表中。最后,我们打印出原始数组以及每个分区的最大值。
注意,这只是一个示例代码,实际上你可以根据自己的需求进行修改和扩展。
上一篇:按照三个字段对Mongo进行分组,获取一个字段的唯一计数。
下一篇:按照三列排序