• 使用Pandas读取大型Excel文件


    import os
    import pandas as pd
    
    HERE = os.path.abspath(os.path.dirname(__file__))
    DATA_DIR = os.path.abspath(os.path.join(HERE, '..', 'data'))
    
    def make_df_from_excel(file_name, nrows):
        """Read from an Excel file in chunks and make a single DataFrame.
        Parameters
        ----------
        file_name : str
        nrows : int
            Number of rows to read at a time. These Excel files are too big,
            so we can't read all rows in one go.
        """
        file_path = os.path.abspath(os.path.join(DATA_DIR, file_name))
        xl = pd.ExcelFile(file_path)
    
        # In this case, there was only a single Worksheet in the Workbook.
        sheetname = xl.sheet_names[0]
    
        # Read the header outside of the loop, so all chunk reads are
        # consistent across all loop iterations.
        df_header = pd.read_excel(file_path, sheetname=sheetname, nrows=1)
        # print(f"Excel file: {file_name} (worksheet: {sheetname})")
        print(f"文件名:{file_name}")
        print(f"工作表:{sheetname}")
    
        chunks = []
        i_chunk = 0
        # The first row is the header. We have already read it, so we skip it.
        skiprows = 1
        while True:
            df_chunk = pd.read_excel(
                file_path, sheetname=sheetname,
                nrows=nrows, skiprows=skiprows, header=None)
            skiprows += nrows
            # When there is no data, we know we can break out of the loop.
            if not df_chunk.shape[0]:
                break
            else:
                # print(f"  - chunk {i_chunk} ({df_chunk.shape[0]} rows)")
                print(f"行数:{df_chunk.shape[0]}")
                chunks.append(df_chunk)
            i_chunk += 1
    
        df_chunks = pd.concat(chunks)
        # Rename the columns to concatenate the chunks with the header.
        columns = {i: col for i, col in enumerate(df_header.columns.tolist())}
        df_chunks.rename(columns=columns, inplace=True)
        df = pd.concat([df_header, df_chunks])
        return df
    
    if __name__ == '__main__':
        df = make_df_from_excel('/Users/mac/Desktop/Data/demo.xlsx', nrows=1000000)
    

    from: cnblogs.com/everfight/p/pandas_read_large_number.html

  • 相关阅读:
    有道翻译js解密(1)
    Python面试题之Python正则表达式re模块
    go语言从例子开始之Example4.常量
    go语言从例子开始之Example3.变量
    go语言从例子开始之Example2.类型
    go语言从例子开始之Example1.helloworld
    python模块打补丁
    gevent协程之猴子补丁带来的坑
    charles抓包小程序
    httptesting HTTP(s)接口自动化测试框架
  • 原文地址:https://www.cnblogs.com/hankleo/p/11681865.html
Copyright © 2020-2023  润新知