2015-05-26 9 views
0

Я хочу просто создать файл csv из построенного DataFrame, поэтому мне не нужно использовать Интернет для доступа к информации. Строки списков в коде:Python: Pandas DataFrame to CSV

Мой текущий код ««Компания»«CIK»» Ticker сектор ««в отрасли»» выглядит следующим образом:

def stockStat(): 
    doc = pq('https://en.wikipedia.org/wiki/List_of_S%26P_500_companies') 

    for heading in doc(".mw-headline:contains('S&P 500 Component Stocks')").parent("h2"): 
     rows = pq(heading).next("table tr") 
     cik = [] 
     ticker = [] 
     coName = [] 
     sector = [] 
     industry = [] 

     for row in rows: 
      tds = pq(row).find("td") 
      cik.append(tds.eq(7).text()) 
      ticker.append(tds.eq(0).text()) 
      coName.append(tds.eq(1).text()) 
      sector.append(tds.eq(3).text()) 
      industry.append(tds.eq(4).text()) 



      d = {'CIK':cik, 'Ticker' : ticker, 'Company':coName, 'Sector':sector, 'Industry':industry} 

     stockData = pd.DataFrame(d) 

     stockData = stockData.set_index('Ticker') 



stockStat() 
+1

так что вы пробовали 'df.to_csv()'? – EdChum

+0

stockData.to_csv()? Я пробовал это, но он завершил процесс, и csv не появился в моей IDE –

+1

да, вы пробовали это? http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.to_csv.html#pandas.DataFrame.to_csv – EdChum

ответ

1

Как EdChum уже упоминалось в комментариях, создание CSV из pandas DataFrame выполняется с помощью метода DataFrame.to_csv().

Метод dataframe.to_csv() принимает множество аргументов, все они описаны в документации метода DataFrame.to_csv(). Вот небольшой пример для вас:

import pandas as pd 

df = pd.DataFrame({'mycolumn': [1,2,3,4]}) 
df.to_csv('~/myfile.csv') 

После этого myfile.csv должен быть доступен в вашем домашнем каталоге. Если вы используете окна, сохранение файла в «C: \ myfile.csv» должно работать лучше как доказательство концепции.

Смежные вопросы