跳转至

190

本文总阅读量次 ,原创教程,严禁转载
@author jackzhenguo
@desc 随机读取文件的K行,生成N个
@tag
@version 
@date 2020/03/28

第190个小例子: 随机读取文件的K行,生成N个

def random_lines_save(filename,gen_file_cnt=10):
    """
    随机选取文件的某些行并保存,想要生成这类文件的个数由参数
    @param: gen_file_cnt 指定 

    @param: filename 读入文件的完整路径
    @param: gen_file_cnt 想要产生的文件个数
    """
    df = pd.read_excel(filename)
    for i in range(gen_file_cnt):
        n = random.randint(1,len(df))
        dfs = df.sample(n)
        dfs.to_excel(str(n)+".xlsx",index=False)
        print(str(n)+".xlsx")

这是一个很实用的函数,用于随机生成K行N个文件,使用场景:原来的文件行数较多,想从中随机提取组合N个文件时。

上一个例子 下一个例子