2015-09-14 50 views
0

我有一個問題應該有一個非常基本的解決方案,但是我一直無法找到它。在Python中使用索引返回與用戶輸入相關的所有值

我有以下代碼可以幫助我提取某些股票的信息。然而,我如何分割數據,以便它只會給我一個特定股票的數據。

例如,我過濾數據,以便我只獲取'TSLA'的股票信息。任何幫助將不勝感激。

import pandas as pd 
from pandas.io.data import DataReader 
from datetime import datetime 

symbols = ['GOOG','AAPL','F','TSLA'] 

df = pd.DataFrame() 

for symbol in symbols: 
    ClosingPrice = DataReader(symbol, 'yahoo', datetime(2015,1,1), datetime(2015,9,1)) 
    ClosingPrice = ClosingPrice.reset_index() 
    ClosingPrice['Symbol'] = symbol 
    df = df.append(ClosingPrice) 

    print (df) 
+0

你想要什麼?像'符號[1]'? –

+0

我希望整個DF數據框被重新打印,但僅限於用戶輸入的股票。因此,就像問題一樣,如果我想過濾TSLA的數據,它將顯示2015年1月1日至2015年9月1日的所有相關信息,例如(開放,低,高等) – Raptor776

回答

0

基本上你可以:

df.loc[df['Symbol'] == 'TSLA'] 

下面是一些useful docs

+0

非常感謝爲此,我還找到了一個解決方案,使用stk_list = ['TSLA'],然後使用「isin」特性df [df ['Symbol']。isin(stk_list)]。 – Raptor776

相關問題