第190个小例子： 随机读取文件的K行，生成N个
编辑日期: 2024-11-28 文章阅读: 次
@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个文件时。
