docker-competition/hello_world.py

import os,sys
import json
import numpy as np

def run():
    res = {"Q1": "Hello world","Q2":0,"Q3":[]}

    # with open(r"/tcdata/num_list.csv") as file:
    data = np.loadtxt(r"/tcdata/num_list.csv", delimiter=",", skiprows=0)
    
    # 求和
    res.update({"Q2": int(sum(data))})
       # res.update({"Q3": [1, 2, 3, 4]})

    # 寻找最大的10个数，从大到小生成一个ListList.
    data2 = np.sort(data, axis=0)[::-1][0:10]
    for i in range(len(data2)):
        res["Q3"].append(int(data2[i]))
    # res.update({"Q3", []})

    with open("result.json","w") as file:
        file.write(json.dumps(res))
    print(json.dumps(res))

if __name__ == "__main__":
    run()