python使用pandas读取excel

Posted 一起来学python

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python使用pandas读取excel相关的知识,希望对你有一定的参考价值。

# -*- coding: utf-8 -*-
import pandas as pd
from collections import defaultdict
import json
from pathlib import Path

DATA = "6.28"
dd = defaultdict(list)


def save_file(l, filename="sss"):
    _temp = Path().cwd()
    file_path = _temp / f"{filename}_list.txt"
    with file_path.open("w", encoding="utf-8") as fs:
        fs.write("
".join(l))


def excel_to_list():
    sExcelFile = "./update_video_task.xlsx"
    df = pd.read_excel(sExcelFile, sheet_name=‘sheet1‘)
    df = df.fillna(method=‘ffill‘)
    df = df.loc[:, ~df.columns.str.contains(‘^Unnamed‘)]
    d = df.to_dict("records")
    return d


def gen_new_list(l, data=DATA):
    for d in l:
        if str(d["data"]) == data:
            name = d["name"]
            source_token = int(d["token_id"])
            mid = d["mid"]
            yield {"name": name, "source_token": int(source_token), "mid": mid}


def create_task_list():
    d = excel_to_list()
    weibo_list = []
    bilibili_list = []
    for item in gen_new_list(d):
        name = item["name"]
        mid = item["mid"]
        source_token = item["source_token"]
        dd[name].append((source_token, mid))
    for k, v in dd.items():
        mid_list = [_id[1] for _id in v]
        source_token = v[0][0]
        item = {"source_token": source_token, "mid_list": mid_list}
        url = f‘http://{{"data":[{json.dumps(item)}]}}‘

        task = f"{k} {url}"
        if k.startswith("zzz"):
            zzz_list.append(task)
        elif k.startswith("sss"):
           sss_list.append(task)

    save_file(ssss_list, filename="sss")
    save_file(zzz_list, filename="zzz")


if __name__ == ‘__main__‘:
    create_task_list()

以上是关于python使用pandas读取excel的主要内容,如果未能解决你的问题,请参考以下文章

如何使用 Pandas 从 Excel 中读取某些列 - Python

python Excel工作表使用pandas读取

pandas读取excel文件出错啥原因?

官方推荐:6种Pandas读取Excel的方法,正确答案都写在源代码里了~太方便了

Python:当文件在列标题中有特殊字符时,使用 Pandas 读取 Excel 文件

pandas读取excel文件指定列