Python Pandas 排序

    import pandas as pd  
      
    security = ['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 = 0  
    for stock in security:  
        score = stockSocre[i]  
        i = i + 1  
        dst_stocks[stock] = score  
    df = 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个股票

Python Pandas 排序

Python, quant | RSS 2.0 | Respond | Trackback |