Python 随机生成100行1000列的0~5之间的数据,并保存到csv文件

这个东西比较简单,所以直接上图,上代码了。

 生成的数据文件如下图所述:

Python 随机生成100行1000列的0~5之间的数据,并保存到csv文件

代码如下:

#-*- coding: UTF-8 -*-

import numpy as np
import pandas as pd
import random
import csv
def main():

    ranList=[]
    my = []
    for i in range(0,100000):
        r=random.randint(0,5)
        ranList.append(r)
    temp = 1
    mx = []
    for i in ranList:
        if temp <=1000:
            mx.append(i)
            temp = temp + 1
        if temp == 1001:
            my.append(mx)
            mx = []
            temp = 1
    rl=pd.DataFrame(my)
    print(rl.shape)
    rl.to_csv("G:\PythonCode\MLself\Data\data.csv",index=False,sep=',')