Data Frame to CSV

user_16978547 0
Viewed 108

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:\Users\etcetc, 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!
python
2 Answers

The df.to_csv is in your for loop so it writes a CSV file at every iteration. However, the name of the CSV file is always the same, so it overwrites the previous file at every iteration.

user_110926360

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_82292650
Related