注:本文一下代码均在jupyter notebook中完成
一、文件读写
1.利用savetxt函数将数据存储到文件
语法:np.savetxt(filename,data)
import numpy as np #导入模块
#准备一个数据 data
arr=np.array(range(6))
arr
np.savetxt('arr.txt',arr) #运行 arr保存到当前文件夹中,运行结果是科学计数法
2.读取csv文件
语法:loadtxt(fname, dtype=, comments=’#’, delimiter=None, converters=None, skiprows=0, usecols=None, unpack=False, ndmin=0, encoding=‘bytes’)
fname: str 读取的CSV文件名
delimiter:str 数据的分割符
usecols:tuple(元组) 执行加载数据文件中的哪些列
unpack:bool(布尔值) 是否将加载的数据拆分为多个组 true拆 false不拆
skipprows:int 跳过多少行 一般用于跳过前几行的描述性文字
encoding:bytes 编码格式
注:主要参数就以上
需求:将CSV文件(如下图)中的收盘价和成交量分别放入数组
#用close这个变量存放收盘价,用amount这个变量来存放成交量
close,amount=np.loadtxt('data.csv',delimiter=',',usecols=(6,7),unpack=True,skiprows=1)
#也可以用负值
#close,amount=np.loadtxt('data.csv