呵呵呵呵呵哈哈：import sys import math def softmax(values): m = max(values) x = [v - m for v in values] x_exp = [math.exp(x) for x in x] x_exp_sum = sum(x_exp) max_exp = max(x_exp) s = max_exp / x_exp_sum idx = x_exp.index(max_exp) return idx, s if __name__ == "__main__": n, k = map(int, sys.stdin.readline().strip().split()) logits = [] for i in range(n): logits.append(list(map(float, sys.stdin.readline().strip().split()))) for logit in logits: idx, s = softmax(logit) print(idx, format(s, '.6f'))