Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from pathlib import Path
- import pandas as pd
- home = Path(r"C:\python\xlsx")
- stock_ids = [1218, 1402, 1444, 1513, 1526, 1536, 1712, 2066, 2231, 2305, 2380,
- 2387, 2413, 2538, 2542, 2850, 3022, 3202, 3296, 3322, 3346, 3680, 3710, 4114,
- 4402, 4743, 5353, 5498, 6279, 6285, 6570, 6605, 8069, 8105, 8403, 9802, 9907]
- a = pd.DataFrame(stock_ids, columns = ["stock_id"])
- b = pd.read_excel(home.joinpath("產業類別.xlsx"))
- data = a.merge(b, on='stock_id')
- category_data = pd.DataFrame({k:v.reset_index(drop=True) for k, v in data.groupby("產業類別")["stock_id"]})
- category_data.to_excel(home.joinpath("output.xlsx"),index= None)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement