Python -- 生成测试表格数据

原创已于 2023-02-17 17:07:54 修改 · 1.1w 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

于 2022-01-19 15:51:36 首次发布

MyCode 专栏收录该内容

32 篇文章

订阅专栏

博客介绍了使用Python生成Excel表格的方法。首先需安装faker和pandas两个第三方库，通过pip install命令即可完成。之后给出代码示例，可生成包含1000条信息的Excel表格。

部署运行你感兴趣的模型镜像

先安装第三方库

pip install faker
pip install pandas

代码示例

下方代码生成excel表格1000条信息

# -*- coding: utf-8 -*-
# @Author   : zbz

import os

from faker import Faker
import pandas as pd


class GenExcel:
    def __init__(self, dp, f, count=10):
        if not os.path.exists(dp):
            os.mkdir(dp)
        self.file_path = "{}/{}".format(dp, f)
        self.count = count
        self.faker = Faker("zh_cn")

    def get_data(self):
        name = self.faker.name()                        # 姓名
        age = self.faker.random.randint(18, 60)         # 年龄
        job = self.faker.job()                          # 工作
        salary = self.faker.random_number(digits=4)     # 工资
        company = self.faker.company()                  # 公司
        address = self.faker.address()                  # 地址
        phone = self.faker.phone_number()               # 电话号码
        ssn = self.faker.ssn()                          # 身份证号码
        letter = self.faker.random_letter()             # 字母
        data = vars()
        del data["self"]
        return data

    def start(self):
        data = [self.get_data() for _ in range(self.count)]
        df = pd.DataFrame(data)
        df.to_excel(self.file_path, index=False)
        print("success ==> {}".format(self.file_path))


if __name__ == '__main__':
    ge = GenExcel(dp="files", f="salry.xlsx", count=1000)
    ge.start()