Python Pandas 排序

来源:互联网 发布:战舰世界闪电数据 编辑:程序博客网 时间:2024/06/15 11:11
import pandas as pdsecurity = ['601318.XSHG','600036.XSHG','000333.XSHE','000651.XSHE','600660.XSHG','600887.XSHG','600004.XSHG','002415.XSHE','600519.XSHG','000002.XSHE','601166.XSHG','601006.XSHG']stockSocre = [1,3,4,2,1,3,1,2,1,5,6,3]dst_stocks = {}i = 0for stock in security:    score = stockSocre[i]    i = i + 1    dst_stocks[stock] = scoredf = pd.DataFrame(dst_stocks.values(), index=dst_stocks.keys())df.columns = ['score']df = df.sort(columns='score', ascending=False)print(df)buyStockList = list(df.index)[:3]print(buyStockList)



选取分数最高的3个股票