zoukankan      html  css  js  c++  java
  • 使用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

  • 相关阅读:
    谁是你随时可以说话的人
    我们在帝都这么拼,为的是什么?
    CVE-2016-4758: UXSS in Safari's showModalDialog
    JSON-SCHEMA
    JS城市data
    linux(centos )mongodb install
    python pip install
    基于chrome内核的UXSS
    Trying to hack Redis via HTTP requests
    Apache Solr 访问权限控制
  • 原文地址:https://www.cnblogs.com/hankleo/p/11681865.html
Copyright © 2011-2022 走看看