python 数据可视化 -- 读取数据

从 CSV 文件中读取数据

import sys
import csv  # python 内置该模块 支持各种CSV文件

file_name = r"..\ch02_data\ch02-data.csv"  # r 指明字符串不用转义
data = []

try:
    with open(file_name) as f:  # 打开文件
        reader = csv.reader(f)  # 获取 reader 对象，通过对该对象的遍历获取文件的所有内容
        header = next(reader)  # 读取文件头
        data = [row for row in reader]  # 读取文件内容，并存放到列表中
except csv.Error as e:
    print("Error reading CSV file at line %s: %s" %(reader.line_num, e))
    sys.exit(-1)

if header:
    print(header)
    print("===========================")

for datarow in data:
    print(datarow)

从 Excel 中读取数据

www.python-excel.org 项目提供的软件包可以方便 python 处理 excel 文件。

该项目提供的 package 有：openpyxl  xlsxwriter  xlrd  xlwt  xlutils 

安装方法：以管理员方式打开 cmd --> pip install xlrd

import xlrd  ## 专门读取 excel 的 package
# from pprint import pprint

file = r"..\ch02_data\ch02-xlsxdata.xlsx"
wb = xlrd.open_workbook(filename = file)  # 读取 excel 文件，返回 xlrd.book.Book 类的实例对象
ws = wb.sheet_by_name("Sheet1")  # 根据 sheet name 读取 sheet 中的数据，返回 xlrd.sheet.Sheet 类的实例对象
data_set = []
for r in range(ws.nrows):  # sheet 的行数
    col = []
    for c in range(ws.ncols):  # sheet 的列数
        col.append(ws.cell(r, c).value)  # 获取指定的 Cell 的实例对象，通过  value 属性获取值
    data_set.append(col)
# pprint(data_set)
for i in data_set:
    print(i)

从定宽数据文件导入数据