关于python读取csv的几种方法
2019-02-27 本文已影响1人
fred_33c7
CSV是一种常见的数据存储文件,在python里面我们应该怎么样去打开并读取它呢?
环境:python3.6 注意本文中的一些方法在python2.7中并不适用需要修改
数据csv的大概样子:
csv文件截图
1. 自己写
# f = open(path,'rt').readlines()
f = open(path,'rt').readlines() # f=file(path) in py2.7
x = []
y = []
for i, d in enumerate(f):
if i == 0:
continue
d = d.strip()
if not d:
continue
print(d)
d = map(float, d.split(','))
d = list(d) # 这一句话在py2.7中不需要
x.append(d[1:-1])
y.append(d[-1])
pprint(x)
pprint(y)
print("--------------")
x = np.array(x)
y = np.array(y)
2. python自带库
f = open(path, 'rt').readlines()
print(f)
d = csv.reader(f)
for line in d:
print(line)
3. numpy读入
p = np.loadtxt(path, delimiter=',', skiprows=1)
print(p)
4. pandas读入 (一般用这种)
data = pd.read_csv(path) # TV、Radio、Newspaper、Sales
x = data[['TV', 'Radio', 'Newspaper']]
# x = data[['TV', 'Radio']]
y = data['Sales']
print(x)
print(y)