Skip to content
Advertisement

Data Frame to CSV

I am currently trying to export my data frame into a csv. The basis of the code is to run a list of tickers through yahoo_fin and get options data. The code works, and I retrieve info for every ticker, however, when I assign the stocks a data frame to export into csv form, the csv only opens with the last item on the list. Example:

stocks = ['AAPL', 'AMZN']

for stock in stocks:

    try:
        pd.set_option('display.max_columns', None)
        chain = options.get_options_chain(stock, 'October 15, 2021')
        print(chain['calls'])
    except:
        ValueError

    df = pd.DataFrame(chain['calls'], columns=['Contract Name', 'Last Trade Date', 'Strike', 'Last Price', 'Bid', 'Ask', 'Change', 'Volume', 'Implied Volatility'])
    df.to_csv(r'C:Usersetcetc, index= False, header= True)```

#I am new to python and any help is appreciated! I tried running the data through XlsxWriter and it didn't work either. 

#Thanks!





Advertisement

Answer

Your df.to_csv is in the for loop, so it overwrites the previously generated file in every iteration, but the behaviour here should be to append the data at the end of your existing CSV file instead of overwriting.

To do this, you can try writing the CSV with mode = 'a' as an argument along with your index and name parameters, and you have to handle the header parameter, as in general, you won’t be needing it FOR the append data. You can find a solution for it here

Alternatively, the preferred way would be to save your data into a dataframe from the loop and write the whole dataframe outside the loop in a single write operation, that way you won’t have to worry about handling the header.

User contributions licensed under: CC BY-SA
4 People found this is helpful
Advertisement