Python 数据分组并计算最小值:按 C、D 分组,取 A 为 1 的 B 列最小值
假设数据存储在一个名为 data 的二维数组中,代码如下:
# 初始化字典
grouped_data = {}
# 遍历数据,按 C,D 分组
for row in data:
key = (row[2], row[3])
if key not in grouped_data:
grouped_data[key] = []
grouped_data[key].append(row)
# 遍历分组后的数据,取 A 为 1 的 B 列的 min() 存为 E 列
for key, group in grouped_data.items():
for row in group:
if row[0] == 1:
min_b = min([r[1] for r in group])
row.append(min_b)
上述代码首先定义了一个字典 grouped_data,用于按 C,D 分组存储数据。然后遍历数据,将相同 C,D 的数据放入同一个分组中。最后再次遍历分组后的数据,对于每个分组中的数据,如果 A 列为 1,则取 B 列的最小值存入 E 列中。最终的结果就是在原数据的基础上添加了一列 E。
原文地址: https://www.cveoy.top/t/topic/n46y 著作权归作者所有。请勿转载和采集!