cok-filter

小工具

# -*- coding: utf-8 -*-
import pandas as pd


def filterExcel(cokpath, keywords):
    df = pd.read_excel(cokpath)
    data_dict = df.to_dict()
    res_dic = {"名字": [], "登記": [], "座標": []}
    for k, v in data_dict["名字"].items():
        if v.startswith(keywords):
            res_dic["名字"].append(v)
            res_dic["登記"].append(data_dict["登記"][k])
            res_dic["座標"].append(data_dict["座標"][k])

    data_frame = pd.DataFrame(res_dic)
    data_frame.to_excel('{0}-INFO.xls'.format(keywords), sheet_name='cok_lgd')


if __name__ == '__main__':
    COKPATH = input("請輸入待處理的Excel文件路徑:")
    ALLIANCE_NAME = input("請輸入須要篩選的聯盟名稱：")
    keywords = "({0})".format(ALLIANCE_NAME)
    filterExcel(COKPATH, keywords)