python 读取CSV文件 #写入数据 #pandas # numpy

import numpy as np
import csv

data = []
for i in range(100):
    x = np.random.uniform(3.,12.)
    eps = np.random.normal(0.,0.1)
    y = 1.477 * x + 0.089 + eps
    data.append([x,y])
data = np.array(data)
data.shape  # 产生（100，2）数据

# 写数据到文件
f = open('data_1.csv','w')
# 获取文件对象
csv_writer = csv.writer(f)
csv_writer.writerow(['1','2'])
for i in range(100):
    csv_writer.writerow(data[i]) # 行写操作
# 关闭文件
f.close()

# pandas 读取CSV 文件
import pandas as pd
data = pd.read_csv('data_1.csv')
# 使用numpy读取数据
data = np.array(data)
data.shape

# numpy 读取CSV文件
path = "data_1.csv"
with open(path) as f:
    data = np.loadtxt(f,delimiter = ',',skiprows=1) # skiprows = n 跳过第 n 行
data = np.array(data)
data.shape