@author jackzhenguo
@desc 随机读取文件的K行,生成N个
@tag
@version
@date 2020/03/28
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个文件时。
[上一个例子](189.md) [下一个例子](191.md)