第二章 了解数据

CSV和excel的区别

csv是纯文本文件，但是excel包含很多格式信息

csv文件体积更小，创建分发读取更加方便，适合存放结构化信息。

 

导入数据代码

从CSV文件导入数据

数据文件和读取数据文件代码在相同目录下。

import csv


filename = 'star98.csv'
data = []  # define an empty list

try:
    with open(filename) as f:
        reader = csv.reader(f)
        header = next(reader)
        data = [row for row in reader]
except csv.Error as e:
    print("Error reading CSV file at line %s: {}" .format(reader.line_num, e))
    sys.exit(-1) #

if header:
    print(header)
    print("======================")
for datarow in data:
    print(datarow)

c

 

从excel表导入数据

import xlrd
file = 'testused.xlsx'
wb = xlrd.open_workbook(filename=file)
ws = wb.sheet_by_name('Sheet1')
dataset = []
for r in range(ws.nrows):
    col = []
    for c in range(ws.ncols):
        col.append(ws.cell(r ,c).value)
    dataset.append(col)

from pprint import pprint
pprint(dataset)

 

import scipy.misc
import matplotlib.pyplot as plt

ascent = scipy.misc.ascent()  # 获取用于测试的图像
plt.gray()  # 显示灰度图

plt.imshow(ascent)  # 显示图像
plt.colorbar()  # 显示色条
plt.show()  # 显示结果