Pandas读取大文件

皮大大

发布于 2021-03-01 13:19:17

2.2K0

发布于 2021-03-01 13:19:17

文章被收录于专栏：机器学习/数据可视化

Pandas技巧-如何读取大文件

本文中记录的是如何利用pandas来读取大文件，4个技巧：

如何利用read_csv函数读取没有表头的文件
get_chunk()方法来分块读取数据
concat()方法将数据库进行叠加（垂直方向）
若数据量过大，采取随机抽放（是否放回）

filepath = open("taobao.csv",errors="ignore")  # 指定文件路径
reader = pd.read_csv(filepath,
                     header=None,
                     names=["user_id","item_id","cat_id","type","time"],  # 指定列属性名称
                     iterator=True)

# loop,chunkSize,chunks = True, 10000000, []  # 连续赋值语句
loop = True
chunkSize = 10000000
chunks = []

while loop:  # loop一直为True，执行循环
    try:
        chunk = reader.get_chunk(chunkSize)
        chunks.append(chunk)
    except StopIteration:
        loop = False
        print("Iteration is stopped.")

# 如果考虑数据量过大，只抽取部分的数据来进行分析，采取不放回抽样的方式
# pd.concat(chunks, ignore_index=True).sample(frac=0.05, replace=False)  不放回抽样、记录不重复

df = pd.concat(chunks, ignore_index=True)

参考文章：https://www.cnblogs.com/frchen/p/5749814.html

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021-1-21，如有侵权请联系 cloudcommunity@tencent.com 删除

数据库