Python 生成随机数

生成指定文件名的随机数

# 生成随机数 
# 指定数据类型：整数(1--100000)、浮点数(五位小数，1--100)
# 指定数据量
# 指定文件名
import random,sys

l = len(sys.argv)
if(l == 4):
    form = sys.argv[1]
    filename = sys.argv[3]
    if(form == "int"):
        n = int(sys.argv[2])
        fp = open(filename,"w")
        for i in range(n-1):
            a = random.randint(1,100000)
            fp.write(str(a)+' ')
        a = random.randint(1,100000)
        fp.write(str(a))
        fp.close()
        print("Success !!")
    elif(form == "float"):
        n = int(sys.argv[2])
        fp = open(filename,"w")
        for i in range(n-1):
            a = round(random.uniform(1.0,100.0),5)
            fp.write(str(a)+' ')
        a = round(random.uniform(1.0,100.0),5)
        fp.write(str(a))
        fp.close()
        print("Success !!")
    else:
        print("Error!!")
else:
    print("Error!!")
    print("Usage: python datagen.py int/float DataNumber filename")

针对spdz，生成随机数到Input-P*-* 。

# 生成随机数 ，只针对两方的数据生成
# 指定数据类型：整数(1--100000)、浮点数(五位小数，1--100)
# 指定数据量
# 指定线程数量

import random,sys

l = len(sys.argv)
if(l == 4):
    form = sys.argv[1]
    threadnum = int(sys.argv[3])
    for tn in range(threadnum+1):
        filename0 = "Input-P0-"+str(tn)
        filename1 = "Input-P1-"+str(tn)
        if(form == "int"):
            n = int(sys.argv[2])
            fp = open(filename0,"w")
            for i in range(n-1):
                a = random.randint(1,100000)
                fp.write(str(a)+' ')
            a = random.randint(1,100000)
            fp.write(str(a))
            fp.close()

            fp = open(filename1,"w")
            for i in range(n-1):
                a = random.randint(1,100000)
                fp.write(str(a)+' ')
            a = random.randint(1,100000)
            fp.write(str(a))
            fp.close()
            print("Success !!")
        elif(form == "float"):
            n = int(sys.argv[2])
            fp = open(filename0,"w")
            for i in range(n-1):
                a = round(random.uniform(1.0,100.0),5)
                fp.write(str(a)+' ')
            a = round(random.uniform(1.0,100.0),5)
            fp.write(str(a))
            fp.close()

            fp = open(filename1,"w")
            for i in range(n-1):
                a = round(random.uniform(1.0,100.0),5)
                fp.write(str(a)+' ')
            a = round(random.uniform(1.0,100.0),5)
            fp.write(str(a))
            fp.close()
            print("Success !!")
        else:
            print("Error!!")
else:
    print("Error!!")
    print("Usage: python datagen.py int/float DataNumber ThreadNumber")