07_加载csv、excel文件

---

博文配套视频课程：24小时实现从零到AI人工智能

---

Pandas文件读取API

pandas提供了一些用于将表格型数据读取为DataFrame对象的函数，其中用的最多的就是read_csv()函数和read_table()

1. pd.read_sql()
2. pd.read_html()
3. pd.read_json()
4. pd.read_excel()
5. pd.read_csv()

读取CSV文件

import numpy as np
import pandas as pd
import random

print('加载本地的csv格式文件')
df = pd.read_csv("../data/groupby.csv")
df.info() # 显示相关的信息
print(df.head(n=3))
# 获取某一列的数据
print(df['Name'])
print(df.values,type(df.values))
# 通过ndarray API也可以直接获取values
print(df.values[:,[1,3]])

读取excel文件

import numpy as np
import pandas as pd
import random

df = pd.read_excel("../data/user.xls",index_col='编号')
df.info()
print(df.head(n=3))
# 随机生成940个1~3之间的数
type = [random.randint(1,3) for _ in range(940)]
# df追加列
df = pd.concat([df,pd.Series(data=type,index=df.index,name='用户类别')],axis=1)
df.to_excel("../data/user_type.xls")