标签：sir csv 多测师 self output 杭州 data company row

from faker import Faker
import csv
import pandas as pd
# from urllib import parse
class Create_Data(object):
    
    def __init__(self):
        # 选择中文
        fake = Faker('zh_CN')
        # 生成数据改变循环体来控制数据量rang(?)
        self.data_total = [
            [fake.name(), fake.job(), fake.company(), fake.phone_number(), fake.company_email(), fake.address(),
             fake.date_time(tzinfo=None)] for x in range(5)]
        print(self.data_total)

    # 写入excel
    # def deal_excel(self):
    #     df = pd.DataFrame(self.data_total,
    #                       columns=['name', 'job', 'company', 'phone_number', 'company_email', 'address', 'date_time'])
    #     # 保存到本地excel
    #     df.to_excel("data_total.xlsx", index=False)
    #     print("Processing completed to excel")

    # 写入txt
    def deal_txt(self):
        with open('data_total.txt', 'w', errors='ignore', encoding='utf-8') as output:
            # output.write('name,job,company,phone_number.company_email.address.date_time\n')
            for row in self.data_total:
                rowtxt = '{},{},{},{},{}'.format(row[0], row[1], row[2], row[3],row[4])
                # print(rowtxt)
                output.write(rowtxt.strip(" "))
                # output.write('\n')
            output.close()
        print("Processing completed to txt")
    # 写入 csv
    def deal_csv(self):
        with open("data_toral.csv",'w',encoding='utf-8') as output:
            csv_writer = csv.writer(output)
            for row in self.data_total:
                csv_writer.writerow(row)
            output.close()
        # f = open("data_toral.csv",'w',encoding='utf-8')
        # csv_writer = csv.writer(f)
        # for i in self.data_total:
        #     csv_writer.writerow(i)


    def write_raw_index(self,filename):
        '''
        """在csv文件中第一行添加索引字段"""
        :return:
        '''
        with open(filename, 'r+', encoding='utf-8') as f:
            content = f.read()
            f.seek(0, 0)
            text = 'name'+ ','+'job'+ ',' + 'company'+ ',' + 'phone_number'+ ',' + 'company_email'+ ',' + 'address'+',' + 'date_time'
            f.write(text + '\n' + content)


if __name__ == '__main__':
    cd = Create_Data()
    cd.deal_txt()
    cd.deal_csv()
    f = "D:\project\hangzhou\dcs15\lesson\data_toral.csv"
    cd.write_raw_index(f)

标签：sir,csv,多测师,self,output,杭州,data,company,row
来源： https://www.cnblogs.com/xiaoshubass/p/16352834.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。