Python读取Excel文件中指定的列数并生成CSV文件

0x00 安装Pandas和OpenPyXL

python 复制代码
pip install pandas

pip install openpyxl

0x01 Encoding

python 复制代码
import os
import pandas as pd
import openpyxl
from openpyxl import load_workbook
import csv


def write_dict_to_csv(dict_data, fileName):
    df = pd.DataFrame.from_dict(dict_data)
    # 将DataFrame写入CSV文件
    df.to_csv(fileName, index=False)


def writeCSV(dict_data, fileName):
    # 打开CSV文件
    with open(fileName, 'w', newline='') as file:
        writer = csv.writer(file)

        # writer.writerow(["key", "value"])
        print("====================================")

        # 工况点字典
        msgPoints_dict = {}
        msgPoints_num = 51

        # 遍历字典
        for key, value in dict_data.items():
            print(value)
            # writer.writerow(value)
            if isinstance(value, list):
                for item in value:
                    # 若表格数据为空则过滤掉
                    if isinstance(item, str) and item.startswith('EP') and item is not None:
                        msgPoints_dict[msgPoints_num] = value
                        # writer.writerow(value)
                        msgPoints_num = msgPoints_num + 1

        print("msgPoints_dict_size: " + str(len(msgPoints_dict)))

        for key, value in msgPoints_dict.items():
            print(key, value, end="\n")
            # 检查键值是否以"EP"开头
            if isinstance(value, list) and value and isinstance(value[2], str) and value[2].startswith("EP"):
                # print(value)

                tmp_list = [str(key)]
                tmp_list.extend(value)
                # 写入CSV文件
                # writer.writerow([key, item])
                writer.writerow(tmp_list)

    pass


def main():
    print("=== Read Excel ===")
    fileName = './doc/data.xlsx'

    # 字典类型
    dict_data = {}
    if os.path.exists(fileName):
        # 当前活跃的Sheel页
        wb = load_workbook(fileName)
        print(wb.active)

        shell = wb.active
        # print(ws['B7'].value)
        # print(ws.cell(7, 2).value)
        # rows = ws.rows
        # print(rows)

        columns_to_print = ['B', 'F', 'O']
        index = 1
        for row in shell.iter_rows():
            tmp_list = []
            for col_idx, cell in enumerate(row, start=1):
                column_letter = openpyxl.utils.get_column_letter(col_idx)
                if column_letter in columns_to_print:
                    tmp_list.append(cell.value)
                    # print(cell.value)

            dict_data[index] = tmp_list
            index = index + 1

    writeCSV(dict_data, "output.csv")
    # write_dict_to_csv(dict_data, "msg_points.csv")
    pass


if __name__ == "__main__":
    main()
相关推荐
coderxiaohan1 分钟前
【C++】类和对象1
java·开发语言·c++
诗句藏于尽头28 分钟前
Django模型与数据库表映射的两种方式
数据库·python·django
阿幸软件杂货间30 分钟前
Office转PDF转换器v1.0.py
开发语言·pdf·c#
智数研析社37 分钟前
9120 部 TMDb 高分电影数据集 | 7 列全维度指标 (评分 / 热度 / 剧情)+API 权威源 | 电影趋势分析 / 推荐系统 / NLP 建模用
大数据·人工智能·python·深度学习·数据分析·数据集·数据清洗
扯淡的闲人40 分钟前
多语言编码Agent解决方案(5)-IntelliJ插件实现
开发语言·python
丑小鸭是白天鹅1 小时前
Kotlin协程详细笔记之切线程和挂起函数
开发语言·笔记·kotlin
moxiaoran57531 小时前
Flask学习笔记(一)
后端·python·flask
sali-tec1 小时前
C# 基于halcon的视觉工作流-章34-环状测量
开发语言·图像处理·算法·计算机视觉·c#
java搬砖工-苤-初心不变1 小时前
基于 lua_shared_dict 的本地内存限流实现
开发语言·junit·lua
charlie1145141911 小时前
Kotlin 的 apply / with / run 详解
开发语言·kotlin·程序设计·面对对象